Update app.py

app.py

@@ -6,90 +6,7 @@ from dotenv import load_dotenv
 import re
 import plotly.graph_objs as go
 
-# 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)}"}
+# ... (previous code remains the same)
 
 def plot_risk(potential_cancer_types):
     if not potential_cancer_types:
@@ -112,34 +29,6 @@ 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:
@@ -172,5 +61,4 @@ iface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
-    
+    iface.launch()