Update accourding to new UI

app.py

@@ -27,7 +27,7 @@ async def health_info(request: Request):
     result.replace("\n/", "<br>")
     return {"response": result}
 
-@app.post("/upload")
+@app.post("/report-analysis")
 async def upload_file(file: UploadFile = File(...)):
     try:
         pdf = file.filename.lower().endswith('.pdf')
@@ -48,6 +48,7 @@ Once the text is extracted, continue with step-by-step medical analysis and retu
     context=content,
     
         )
+        
         print(result.final_output)
         return {"result": result.final_output.model_dump()}
     except Exception as e:

chatbot.py

@@ -1,11 +1,20 @@
 # chatbot.py
-from agents import Agent,RunConfig,Runner,OpenAIChatCompletionsModel,AsyncOpenAI,set_tracing_disabled
-import asyncio
+from agents import (Agent,
+                    RunConfig,
+                    Runner,
+                    OpenAIChatCompletionsModel,
+                    AsyncOpenAI,
+                    model_settings,
+                    function_tool,
+                    set_tracing_disabled,
+                    enable_verbose_stdout_logging)
 from dotenv import load_dotenv
 import os
-
+import requests
+import xmltodict
+import json
 set_tracing_disabled(disabled=True)
-
+enable_verbose_stdout_logging()
 
 load_dotenv()
 api_key = os.getenv("GEM_API_KEY")
@@ -26,6 +35,15 @@ config = RunConfig(
     tracing_disabled=True
 )
 
+@function_tool
+async def get_info_about_health(query:str) -> str:
+    """Fetch health information from web based on the query.
+       That helps to provide accurate medical advice."""
+    url = f"https://wsearch.nlm.nih.gov/ws/query?db=healthTopics&term={query}"
+    responce = requests.get(url)
+
+    return responce.text
+
 agent: Agent = Agent(
     name="Doctor",
     instructions="""
@@ -118,14 +136,13 @@ Agar dard barhta hai toh please doctor ko dikhayein.\n
 Take care! ❤️
 ”
 """,
-
+    tools=[get_info_about_health],
+    model_settings=model_settings.ModelSettings(tool_choice="required"),
     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)
-    return result.final_output
-
+    return result.final_output

reportanalysis.py

@@ -15,7 +15,8 @@ from typing import List, Literal
 from agents import (
         Agent,
         AsyncOpenAI, 
-	      OpenAIChatCompletionsModel, 
+          OpenAIChatCompletionsModel, 
+        AgentOutputSchema,
         AgentOutputSchemaBase,
         enable_verbose_stdout_logging,
         set_tracing_disabled
@@ -30,31 +31,30 @@ anonymizer = AnonymizerEngine()
 
 API = os.getenv("GEM_API_KEY")
 
-class TestItem(BaseModel):
-    name: str
-    user_value: str
-    normal_range: str
-    analysis: str | None = None
-    flag: Literal["Red", "Yellow", "Green"]
-
-class ReportSection(BaseModel):
-    title: str
-    tests: List[TestItem]
-    section_summary: str
-
-class AiTipSection(BaseModel):
-    title: str
-    risk: str
-    tips: str
-    action: str
-    diet_suggestion: List[str] = Field(default_factory=list)
-    life_style: List[str] = Field(default_factory=list)
-
-class ReportSchema(BaseModel):
-    report_summary_title: str
-    ai_tip_title: str
-    report_sections: List[ReportSection]
-    ai_tip_sections: List[AiTipSection]
+class AiInsights(BaseModel):
+    overallAssessment: str
+    keyHighlights: List[dict[str, str]]
+    dietaryRecommendations: List[str]
+    lifestyleAdvice: List[str]
+    precautions: List[str]
+    risks: List[str]
+    actions: List[dict[str, str]]
+    tips: List[str]
+
+class KeyFinding(BaseModel):
+    test: str
+    value: int
+    unit: str
+    range: str
+    shortExplaination: str
+    status: Literal["Red", "Yellow", "Green"]
+
+class AnalysisResult(BaseModel):
+  fileName: str
+  reportType: str
+  summary: str
+  keyFindings: List[KeyFinding]
+  aiInsights: AiInsights
 
 client = AsyncOpenAI(
        api_key = API,
@@ -120,14 +120,10 @@ Your Main Task:
 4. Provide a clear summary of the findings.
 5. Offer relevant AI-driven health tips, highlight potential risks, and suggest dietary and lifestyle improvements.
 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}}}
-
+Response format: {'type': 'json_schema', 'json_schema': {'name': 'final_output', 'strict': False, 'schema': {'$defs': {'AiInsights': {'properties': {'overallAssessment': {'title': 'Overallassessment', 'type': 'string'}, 'keyHighlights': {'items': {'additionalProperties': {'type': 'string'}, 'type': 'object'}, 'title': 'Keyhighlights', 'type': 'array'}, 'dietaryRecommendations': {'items': {'type': 'string'}, 'title': 'Dietaryrecommendations', 'type': 'array'}, 'lifestyleAdvice': {'items': {'type': 'string'}, 'title': 'Lifestyleadvice', 'type': 'array'}, 'precautions': {'items': {'type': 'string'}, 'title': 'Precautions', 'type': 'array'}, 'risks': {'items': {'type': 'string'}, 'title': 'Risks', 'type': 'array'}, 'actions': {'items': {'additionalProperties': {'type': 'string'}, 'type': 'object'}, 'title': 'Actions', 'type': 'array'}, 'tips': {'items': {'type': 'string'}, 'title': 'Tips', 'type': 'array'}}, 'required': ['overallAssessment', 'keyHighlights', 'dietaryRecommendations', 'lifestyleAdvice', 'precautions', 'risks', 'actions', 'tips'], 'title': 'AiInsights', 'type': 'object'}, 'KeyFinding': {'properties': {'test': {'title': 'Test', 'type': 'string'}, 'value': {'title': 'Value', 'type': 'integer'}, 'unit': {'title': 'Unit', 'type': 'string'}, 'range': {'title': 'Range', 'type': 'string'}, 'status': {'enum': ['Red', 'Yellow', 'Green'], 'title': 'Status', 'type': 'string'}}, 'required': ['test', 'value', 'unit', 'range', 'status'], 'title': 'KeyFinding', 'type': 'object'}}, 'properties': {'fileName': {'title': 'Filename', 'type': 'string'}, 'reportType': {'title': 'Reporttype', 'type': 'string'}, 'summary': {'title': 'Summary', 'type': 'string'}, 'keyFindings': {'items': {'$ref': '#/$defs/KeyFinding'}, 'title': 'Keyfindings', 'type': 'array'}, 'aiInsights': {'$ref': '#/$defs/AiInsights'}}, 
+'required': ['fileName', 'reportType', 'summary', 'keyFindings', 'aiInsights'], 'title': 'AnalysisResult', 'type': 'object'}}}
 """,
         model = agent_model,
-        output_type=  ReportSchema,
+        output_type= AgentOutputSchema(AnalysisResult, strict_json_schema=False)
 )
 
-
-
-