Update app.py

app.py

@@ -1,50 +1,106 @@
 import gradio as gr
 import os
 from groq import Groq
+import json
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
 
 # Initialize Groq client
 client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
-def get_diagnosis(symptoms):
+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 symptoms, provide a preliminary analysis of potential cancer risks. 
+    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:
-    1. Potential cancer types that might be associated with these symptoms
-    2. Recommendations for further medical tests or consultations
-    3. General health advice
+    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"
+    }}
     
-    Remember to emphasize the importance of consulting with a medical professional for accurate diagnosis and treatment.
+    Ensure the response emphasizes the importance of consulting with a medical professional for accurate diagnosis and treatment.
     """
     
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "user",
-                "content": prompt,
-            }
-        ],
-        model="mixtral-8x7b-32768",
-        temperature=0.5,
-        max_tokens=1000,
-    )
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="mixtral-8x7b-32768",
+            temperature=0.5,
+            max_tokens=1500,
+        )
+        
+        response = json.loads(chat_completion.choices[0].message.content)
+        return response
+    except Exception as e:
+        return {"error": str(e)}
+
+def format_output(response):
+    if "error" in response:
+        return f"An error occurred: {response['error']}"
     
-    return chat_completion.choices[0].message.content
+    output = "Cancer Risk Assessment Results\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"
+    
+    output += "Recommended Tests:\n"
+    for test in response["recommended_tests"]:
+        output += f"- {test['name']}: {test['description']}\n\n"
+    
+    output += f"General Advice:\n{response['general_advice']}\n\n"
+    output += f"DISCLAIMER:\n{response['disclaimer']}"
+    
+    return output
+
+def process_input(age, gender, symptoms, medical_history):
+    diagnosis = get_diagnosis(age, gender, symptoms, medical_history)
+    return format_output(diagnosis)
 
 # Create Gradio interface
 iface = gr.Interface(
-    fn=get_diagnosis,
-    inputs=gr.Textbox(lines=5, label="Enter symptoms (separated by commas)"),
-    outputs="text",
+    fn=process_input,
+    inputs=[
+        gr.Number(label="Age"),
+        gr.Radio(["Male", "Female", "Other"], label="Gender"),
+        gr.Textbox(lines=3, label="Symptoms (separated by commas)"),
+        gr.Textbox(lines=3, label="Relevant Medical History")
+    ],
+    outputs=gr.Textbox(label="Assessment Results", lines=10),
     title="Cancer Risk Assessment Tool",
-    description="This tool provides a preliminary analysis based on symptoms. It is not a substitute for professional medical advice, diagnosis, or treatment.",
+    description="This tool provides a preliminary analysis based on provided information. It is not a substitute for professional medical advice, diagnosis, or treatment.",
+    article="IMPORTANT: This application is for educational purposes only. Always consult with a qualified healthcare provider for medical concerns. The information provided by this tool should not be used for self-diagnosis or treatment."
 )
 
-# Launch the app
-iface.launch()
+# Add usage tracking (for demonstration purposes only, ensure compliance with privacy laws in practice)
+iface.launch(analytics_enabled=True)