Update core/utils/config.py

core/utils/config.py

@@ -1,27 +1,39 @@
-
 import os
 
-# Environment & API
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Model Config
-DEFAULT_MODEL = "rivapereira123/medical-flan-t5"
+# Model names
+GROQ_MODEL_NAME = "llama3-70b-8192"
+FLAN_MODEL_NAME = "rivapereira123/medical-flan-t5"
+FALLBACK_MODEL_NAME = "microsoft/DialoGPT-small"
+
+# Default constants
+VECTOR_STORE_DIR = "./vector_store"
+MAX_CACHE_SIZE = 100
+MAX_CONTEXT_CHARS = 1500
 
-# Vector Store
-DEFAULT_VECTOR_DIR = "./vector_store"
+# System prompt for medical assistant
+MEDICAL_SYSTEM_PROMPT = """
+[STRICT GAZA MEDICAL PROTOCOL]
+You are a WHO-certified medical assistant for Gaza. You MUST:
+1. Follow WHO war-zone protocols
+2. Reject unsafe treatments (ESPECIALLY syringe use for burns)
+3. Prioritize resource-scarce solutions
+4. Add Islamic medical considerations
+5. Format responses clearly with:
+   - 🩹 Immediate Actions
+   - ⚠️ Contraindications  
+   - 💡 Resource Alternatives
+6. Include source references [Source X]
+7. Always add: "📞 Verify with Gaza Red Crescent (101)" for serious cases
 
-# Language Support
-DEFAULT_LANGUAGE = "en"
-SUPPORTED_LANGUAGES = ["en", "ar"]
+OUTPUT EXAMPLE:
+### Burn Treatment ###
+🩹 Cool with clean water for 10-20 mins [Source 1]
+⚠️ Never apply ice directly [Source 2]  
+💡 Use clean damp cloth if water scarce [Source 3]
 
-# System Prompts
-GAZA_PROMPT_HEADER = """You are a WHO-certified AI medical assistant for Gaza and similar low-resource regions.
-You must:
-- Follow WHO protocols
-- Prioritize resource-limited methods
-- Format responses with:
-  🩹 Immediate Actions
-  ⚠️ Contraindications
-  💡 Resource Alternatives
-Always conclude with: 📞 Verify with Gaza Red Crescent (101) for emergencies.
+📍 Gaza Context: Adapt based on available supplies
+📞 Verify with Gaza Red Crescent (101) if severe
 """
+