add chat bot

app.py

@@ -1,9 +1,9 @@
-from fastapi import FastAPI, UploadFile , File
+from fastapi import FastAPI, UploadFile , File , Request
 from fastapi.middleware.cors import CORSMiddleware
 from agents import Runner
-from reportanalysis import Report_Agent, extract_text
 from logging import getLogger
-
+from reportanalysis import Report_Agent, extract_text
+from chatbot import get_health_response 
 app = FastAPI()
 log = getLogger()
 
@@ -19,6 +19,13 @@ app.add_middleware(
 async def root():
     return {"message": "Welcome to the Medical Report Analysis API"}
 
+@app.post("/chatbot")
+async def health_info(request: Request):
+    data = await request.json()
+    msg = data.get("message")
+    result = await get_health_response(msg)  # <- await here
+    return {"response": result}
+
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
     try:

chatbot.py

@@ -0,0 +1,82 @@
+# chatbot.py
+from agents import Agent,RunConfig,Runner,OpenAIChatCompletionsModel,AsyncOpenAI,set_tracing_disabled
+import asyncio
+from dotenv import load_dotenv
+import os
+
+set_tracing_disabled(disabled=True)
+
+
+load_dotenv()
+api_key = os.getenv("GEM_API_KEY")
+
+external_client = AsyncOpenAI(
+    api_key=api_key,
+    base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+)
+
+model = OpenAIChatCompletionsModel(
+    model="gemini-2.0-flash",
+    openai_client=external_client
+)
+
+config = RunConfig(
+    model=model,
+    model_provider=external_client,
+    tracing_disabled=True
+)
+
+agent: Agent = Agent(
+    name="Doctor",
+    instructions="""
+You are **DrXpert**, an intelligent and reliable AI medical assistant designed to help users understand health symptoms and get general guidance.  
+Your job is to **analyze user symptoms**, explain possible causes, and suggest **safe, verified, and general health advice** — without diagnosing or prescribing prescription-only medicines.
+
+🧠 **Core Rules (Follow Strictly):**
+1. Always stay within **medical accuracy** and **safe general advice**.
+2. Never guess, make up, or invent diseases, symptoms, or medicines.  
+   - If you are not 100% sure, respond:  
+     “I’m not completely sure about that. It’s best to consult a real doctor for proper diagnosis.”
+3. Never give harmful or unverified advice, never recommend antibiotics or injections.
+4. If the question is unrelated to health, reply:  
+   “I’m designed to discuss health-related topics only. Please ask something related to symptoms, medicines, or wellness.”
+
+💬 **Response Format (Always Follow This):**
+When the user mentions symptoms (e.g., “I have a fever”, “I feel dizzy”, “I have chest pain”), respond with:
+1. **Possible Common Causes** (1–2 short lines, simple explanation only)
+2. **Safe Medicine Suggestion (Optional)**  
+   - Only general medicines like *Paracetamol, ORS, Panadol, etc.*  
+   - Never mention strong or prescription drugs.
+3. **Precautions (2–3 short, clear points)**  
+   - E.g., “Rest well”, “Drink plenty of fluids”, “Avoid cold drinks”
+4. **Home Remedies (1–2 simple tips)** in **Urdu or English**  
+   - E.g., “Take steam”, “Honey with lemon water”, “Use salt water gargle”
+5. End with a **kind, short line** such as:  
+   - “Allah sehat de ❤️”  
+   - “Khush raho, duaon mein yaad rakhna 😊”
+
+💡 **Tone & Style:**
+- Be warm, calm, caring — like a friendly online doctor.  
+- Keep sentences short, simple, and free of medical jargon.  
+- Use a balance of Urdu & English when possible.  
+- Never repeat questions or unnecessary details.
+
+🚫 **Important Safety Rules:**
+- Do NOT give specific diagnoses.
+- Do NOT discuss suicide, sexual content, or emergencies — instead say:
+  “This seems serious, please visit a nearby hospital immediately.”
+- Do NOT reveal private or personal information.
+
+You are now ready to respond as **DrXpert**, a kind, factual, and safe virtual health guide.
+""",
+
+    model=model
+)
+
+
+async def get_health_response(user_message: str) -> str:
+    print("Running agent with message:", user_message)
+    result = await Runner.run(agent, user_message, run_config=config)
+    print("Final output:", result.final_output)
+    return result.final_output
+

reportanalysis.py

@@ -119,8 +119,9 @@ Your Main Task:
    - Green: Normal or safe
 4. Provide a clear summary of the findings.
 5. Offer relevant AI-driven health tips, highlight potential risks, and suggest dietary and lifestyle improvements.
-
-Return your final response STRICTLY in the JSON structure.
+6. Structure the output in the specified JSON format.
+Response format: {'type': 'json_schema', 'json_schema': {'name': 'final_output', 'strict': True, 'schema': {'$defs': {'AiTipSection': {'properties': {'title': {'title': 'Title', 'type': 'string'}, 'risk': {'title': 'Risk', 'type': 'string'}, 'tips': {'title': 'Tips', 'type': 'string'}, 'action': {'title': 'Action', 'type': 'string'}, 'diet_suggestion': {'items': {'type': 'string'}, 'title': 'Diet Suggestion', 'type': 'array'}, 'life_style': {'items': {'type': 'string'}, 'title': 'Life Style', 'type': 'array'}}, 'required': ['title', 'risk', 'tips', 'action', 'diet_suggestion', 'life_style'], 'title': 
+'AiTipSection', 'type': 'object', 'additionalProperties': False}, 'ReportSection': {'properties': {'title': {'title': 'Title', 'type': 'string'}, 'tests': {'items': {'$ref': '#/$defs/TestItem'}, 'title': 'Tests', 'type': 'array'}, 'section_summary': {'title': 'Section Summary', 'type': 'string'}}, 'required': ['title', 'tests', 'section_summary'], 'title': 'ReportSection', 'type': 'object', 'additionalProperties': False}, 'TestItem': {'properties': {'name': {'title': 'Name', 'type': 'string'}, 'user_value': {'title': 'User Value', 'type': 'string'}, 'normal_range': {'title': 'Normal Range', 'type': 'string'}, 'analysis': {'anyOf': [{'type': 'string'}, {'type': 'null'}], 'title': 'Analysis'}, 'flag': {'enum': ['Red', 'Yellow', 'Green'], 'title': 'Flag', 'type': 'string'}}, 'required': ['name', 'user_value', 'normal_range', 'analysis', 'flag'], 'title': 'TestItem', 'type': 'object', 'additionalProperties': False}}, 'properties': {'report_summary_title': {'title': 'Report Summary Title', 'type': 'string'}, 'ai_tip_title': {'title': 'Ai Tip Title', 'type': 'string'}, 'report_sections': {'items': {'$ref': '#/$defs/ReportSection'}, 'title': 'Report Sections', 'type': 'array'}, 'ai_tip_sections': {'items': {'$ref': '#/$defs/AiTipSection'}, 'title': 'Ai Tip Sections', 'type': 'array'}}, 'required': ['report_summary_title', 'ai_tip_title', 'report_sections', 'ai_tip_sections'], 'title': 'ReportSchema', 'type': 'object', 'additionalProperties': False}}}
 
 """,
         model = agent_model,