Update app.py

app.py

@@ -6,7 +6,87 @@ from dotenv import load_dotenv
 import re
 import plotly.graph_objs as go
 
-# ... (previous code remains the same)
+# Load environment variables
+load_dotenv()
+
+# Initialize Groq client
+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. 
+    This is not a definitive diagnosis and should not replace professional medical advice.
+    
+    Patient Information:
+    Age: {age}
+    Gender: {gender}
+    Symptoms: {symptoms}
+    Medical History: {medical_history}
+    
+    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.
+    """
+    
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama-3.1-8b-instant",  # Updated to a valid Groq model
+            temperature=0.5,
+            max_tokens=1500,
+        )
+        
+        response_content = chat_completion.choices[0].message.content
+        
+        try:
+            response = json.loads(response_content)
+        except json.JSONDecodeError:
+            response = parse_non_json_response(response_content)
+        
+        return response
+    except Exception as e:
+        return {"error": f"An error occurred while communicating with the API: {str(e)}"}
 
 def plot_risk(potential_cancer_types):
     if not potential_cancer_types:
@@ -29,6 +109,34 @@ def plot_risk(potential_cancer_types):
 
     return fig
 
+def format_output(response):
+    if "error" in response:
+        return f"Error: {response['error']}"
+
+    output = "HealthScan AI: Personalized Cancer Risk Insights\n\n"
+    
+    output += "Potential Cancer Types:\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.get("recommended_tests", []):
+        output += f"- {test.get('name', 'N/A')}: {test.get('description', 'No description provided.')}\n\n"
+    
+    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
+
+def validate_input(age, gender, symptoms, medical_history):
+    errors = []
+    if not (0 < age < 120):
+        errors.append("Please enter a valid age between 1 and 120.")
+    if not symptoms.strip():
+        errors.append("Please enter at least one symptom.")
+    return errors
+
 def process_input(age, gender, symptoms, medical_history):
     errors = validate_input(age, gender, symptoms, medical_history)
     if errors: