Update app.py

app.py

@@ -6,6 +6,16 @@ from fastapi.middleware.cors import CORSMiddleware
 from utils.vector_search import get_similar_symptoms
 from utils.sql_queries import get_diseases_by_symptoms, get_related_symptoms
 import sqlite3
+import os
+import httpx
+import json
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+# Get Groq API key from environment variables
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 
 app = FastAPI()
 
@@ -66,3 +76,46 @@ def search(symptom: str, selected: list[str] = Query([])):
         "related_symptoms": suggestions,
         "semantic_matches": similar
     }
+
+@app.get("/disease-info")
+async def disease_info(name: str):
+    """
+    Get detailed information about a disease using Groq's LLM API
+    """
+    if not GROQ_API_KEY:
+        return {"info": "<p>Error: Groq API key not found in environment variables.</p>"}
+    
+    try:
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                "https://api.groq.com/openai/v1/chat/completions",
+                headers={
+                    "Authorization": f"Bearer {GROQ_API_KEY}",
+                    "Content-Type": "application/json"
+                },
+                json={
+                    "model": "llama3-70b-8192",
+                    "messages": [
+                        {
+                            "role": "system",
+                            "content": "You are a helpful medical assistant. Provide accurate, well-structured information about diseases. Format your response in HTML paragraphs with appropriate headings using <h4> tags. Include symptoms, causes, treatments, and prevention when possible. Keep responses concise but informative."
+                        },
+                        {
+                            "role": "user",
+                            "content": f"Provide detailed information about {name}, including symptoms, causes, diagnosis, treatments, and prevention methods when applicable."
+                        }
+                    ],
+                    "max_tokens": 1500,
+                    "temperature": 0.5
+                }
+            )
+            
+            if response.status_code == 200:
+                data = response.json()
+                content = data["choices"][0]["message"]["content"]
+                return {"info": content}
+            else:
+                return {"info": f"<p>Error: Failed to fetch information. Status code: {response.status_code}</p>"}
+                
+    except Exception as e:
+        return {"info": f"<p>Error: {str(e)}</p>"}