Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 from groq import Groq
 import json
 from dotenv import load_dotenv
+import re
 
 # Load environment variables
 load_dotenv()
@@ -12,6 +13,33 @@ client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
+def parse_non_json_response(text):
+    # Attempt to extract structured information from non-JSON text
+    cancer_types = re.findall(r"(?:cancer type|Cancer Type):\s*(.+?)(?:\n|$)", text, re.IGNORECASE)
+    risk_levels = re.findall(r"(?:risk level|Risk Level):\s*(.+?)(?:\n|$)", text, re.IGNORECASE)
+    descriptions = re.findall(r"(?:description|Description):\s*(.+?)(?:\n|$)", text, re.IGNORECASE)
+    
+    potential_cancer_types = [
+        {"name": ct, "risk_level": rl, "description": desc}
+        for ct, rl, desc in zip(cancer_types, risk_levels, descriptions)
+    ]
+    
+    recommended_tests = re.findall(r"(?:Test Name):\s*(.+?)\n(?:description|Description):\s*(.+?)(?:\n|$)", text, re.IGNORECASE)
+    recommended_tests = [{"name": name, "description": desc} for name, desc in recommended_tests]
+    
+    general_advice = re.search(r"(?:General Advice|general advice):\s*(.+?)(?:\n|$)", text, re.DOTALL | re.IGNORECASE)
+    general_advice = general_advice.group(1) if general_advice else "No general advice provided."
+    
+    disclaimer = re.search(r"(?:DISCLAIMER|Disclaimer):\s*(.+?)(?:\n|$)", text, re.DOTALL | re.IGNORECASE)
+    disclaimer = disclaimer.group(1) if disclaimer else "No disclaimer provided. This tool is for educational purposes only and should not replace professional medical advice."
+    
+    return {
+        "potential_cancer_types": potential_cancer_types,
+        "recommended_tests": recommended_tests,
+        "general_advice": general_advice,
+        "disclaimer": disclaimer
+    }
+
 def get_diagnosis(age, gender, symptoms, medical_history):
     prompt = f"""
     Given the following patient information, provide a preliminary analysis of potential cancer risks. 
@@ -23,24 +51,19 @@ def get_diagnosis(age, gender, symptoms, medical_history):
     Symptoms: {symptoms}
     Medical History: {medical_history}
     
-    Please provide a JSON response with the following structure:
-    {{
-        "potential_cancer_types": [
-            {{
-                "name": "Cancer Type",
-                "risk_level": "Low/Medium/High",
-                "description": "Brief description of why this cancer type is considered"
-            }}
-        ],
-        "recommended_tests": [
-            {{
-                "name": "Test Name",
-                "description": "Brief description of why this test is recommended"
-            }}
-        ],
-        "general_advice": "General health advice for the patient",
-        "disclaimer": "A strong disclaimer about the limitations of this assessment"
-    }}
+    Please provide a response with the following structure:
+    Potential Cancer Types:
+    - Cancer Type: [Name]
+      Risk Level: [Low/Medium/High]
+      Description: [Brief description of why this cancer type is considered]
+
+    Recommended Tests:
+    - Test Name: [Name]
+      Description: [Brief description of why this test is recommended]
+
+    General Advice: [General health advice for the patient]
+
+    DISCLAIMER: [A strong disclaimer about the limitations of this assessment]
     
     Ensure the response emphasizes the importance of consulting with a medical professional for accurate diagnosis and treatment.
     """
@@ -60,15 +83,11 @@ def get_diagnosis(age, gender, symptoms, medical_history):
         
         response_content = chat_completion.choices[0].message.content
         
-        # Try to parse the JSON response
+        # Try to parse the response as JSON, if it fails, parse it as text
         try:
             response = json.loads(response_content)
         except json.JSONDecodeError:
-            # If JSON parsing fails, return the raw text response
-            return {
-                "error": "Failed to parse API response as JSON. Raw response:",
-                "raw_response": response_content
-            }
+            response = parse_non_json_response(response_content)
         
         return response
     except Exception as e:
@@ -76,10 +95,7 @@ def get_diagnosis(age, gender, symptoms, medical_history):
 
 def format_output(response):
     if "error" in response:
-        if "raw_response" in response:
-            return f"Error: {response['error']}\n\nRaw API Response:\n{response['raw_response']}"
-        else:
-            return f"Error: {response['error']}"
+        return f"Error: {response['error']}"
     
     output = "HealthScan AI: Personalized Cancer Risk Insights\n\n"