Update app.py

app.py

@@ -13,7 +13,6 @@ client = Groq(
 )
 
 def get_diagnosis(age, gender, symptoms, medical_history):
-    # IMPORTANT: This is a simplified example and should not be used for actual medical diagnosis
     prompt = f"""
     Given the following patient information, provide a preliminary analysis of potential cancer risks. 
     This is not a definitive diagnosis and should not replace professional medical advice.
@@ -59,27 +58,42 @@ def get_diagnosis(age, gender, symptoms, medical_history):
             max_tokens=1500,
         )
         
-        response = json.loads(chat_completion.choices[0].message.content)
+        response_content = chat_completion.choices[0].message.content
+        
+        # Try to parse the JSON response
+        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
+            }
+        
         return response
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": f"An error occurred while communicating with the API: {str(e)}"}
 
 def format_output(response):
     if "error" in response:
-        return f"An error occurred: {response['error']}"
+        if "raw_response" in response:
+            return f"Error: {response['error']}\n\nRaw API Response:\n{response['raw_response']}"
+        else:
+            return f"Error: {response['error']}"
     
-    output = "Vers3Dynamics HealthScan: Personalized Cancer Risk Insights\n\n"
+    output = "HealthScan AI: Personalized Cancer Risk Insights\n\n"
     
     output += "Potential Cancer Types:\n"
-    for cancer in response["potential_cancer_types"]:
-        output += f"- {cancer['name']} (Risk Level: {cancer['risk_level']})\n  {cancer['description']}\n\n"
+    for cancer in response.get("potential_cancer_types", []):
+        output += f"- {cancer.get('name', 'N/A')} (Risk Level: {cancer.get('risk_level', 'N/A')})\n"
+        output += f"  {cancer.get('description', 'No description provided.')}\n\n"
     
     output += "Recommended Tests:\n"
-    for test in response["recommended_tests"]:
-        output += f"- {test['name']}: {test['description']}\n\n"
+    for test in response.get("recommended_tests", []):
+        output += f"- {test.get('name', 'N/A')}: {test.get('description', 'No description provided.')}\n\n"
     
-    output += f"General Advice:\n{response['general_advice']}\n\n"
-    output += f"DISCLAIMER:\n{response['disclaimer']}"
+    output += f"General Advice:\n{response.get('general_advice', 'No general advice provided.')}\n\n"
+    output += f"DISCLAIMER:\n{response.get('disclaimer', 'No disclaimer provided. This tool is for educational purposes only and should not replace professional medical advice.')}"
     
     return output