Update app.py

app.py

@@ -4,11 +4,15 @@ from fpdf import FPDF
 import os
 from groq import Groq  
 from deep_translator import GoogleTranslator  
+import time
 
 # Load API Key
 groq_api_key = os.getenv("groq_api_key")
 groq_client = Groq(api_key=groq_api_key)  
 
+# Load Model
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
 # Translation Function
 def translate_text(text, target_lang="en"):
     try:
@@ -16,43 +20,49 @@ def translate_text(text, target_lang="en"):
     except Exception as e:
         return f"Translation Error: {str(e)}"
 
-# Load Model
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
-
-# Medical Conditions & Specialist Mapping
-conditions = {
-    "Asthma": "Pulmonologist",
-    "COPD": "Pulmonologist",
-    "Pneumonia": "Pulmonologist",
-    "Tuberculosis": "Infectious Disease Specialist",
-    "COVID-19": "Infectious Disease Specialist",
-    "Heart Failure": "Cardiologist",
-    "Hypertension": "Cardiologist",
-    "Diabetes Type 1": "Endocrinologist",
-    "Diabetes Type 2": "Endocrinologist",
-    "Migraine": "Neurologist",
-    "Gastroenteritis": "Gastroenterologist",
-    "Anemia": "Hematologist",
-    "Depression": "Psychiatrist",
-    "Anxiety Disorder": "Psychiatrist",
-    "Chronic Kidney Disease": "Nephrologist",
-    "UTI": "Urologist",
-    "Osteoporosis": "Orthopedic Specialist",
-    "Psoriasis": "Dermatologist",
-    "Epilepsy": "Neurologist"
+# Medical Conditions
+conditions = [
+    "Asthma", "COPD", "Pneumonia", "Tuberculosis", "COVID-19", "Bronchitis",
+    "Heart Failure", "Hypertension", "Diabetes Type 1", "Diabetes Type 2", 
+    "Migraine", "Gastroenteritis", "Anemia", "Depression", "Anxiety Disorder",
+    "Chronic Kidney Disease", "UTI", "Osteoporosis", "Psoriasis", "Epilepsy"
+]
+
+# Specialist Mapping
+specialist_mapping = {
+    "Asthma": ("Pulmonologist", "Respiratory System"),
+    "COPD": ("Pulmonologist", "Respiratory System"),
+    "Pneumonia": ("Pulmonologist", "Respiratory System"),
+    "Tuberculosis": ("Infectious Disease Specialist", "Respiratory System"),
+    "COVID-19": ("Infectious Disease Specialist", "Immune System"),
+    "Heart Failure": ("Cardiologist", "Cardiovascular System"),
+    "Hypertension": ("Cardiologist", "Cardiovascular System"),
+    "Diabetes Type 1": ("Endocrinologist", "Endocrine System"),
+    "Diabetes Type 2": ("Endocrinologist", "Endocrine System"),
+    "Migraine": ("Neurologist", "Nervous System"),
+    "Gastroenteritis": ("Gastroenterologist", "Digestive System"),
+    "Anemia": ("Hematologist", "Blood Disorders"),
+    "Depression": ("Psychiatrist", "Mental Health"),
+    "Anxiety Disorder": ("Psychiatrist", "Mental Health"),
+    "Chronic Kidney Disease": ("Nephrologist", "Urinary System"),
+    "UTI": ("Urologist", "Urinary System"),
+    "Osteoporosis": ("Orthopedic Specialist", "Musculoskeletal System"),
+    "Psoriasis": ("Dermatologist", "Skin Disorders"),
+    "Epilepsy": ("Neurologist", "Nervous System")
 }
 
 # Function to generate expert analysis
 def generate_expert_analysis(condition, symptoms):
-    specialist = conditions.get(condition, "General Physician")
+    specialist_title = specialist_mapping.get(condition, ("General Physician", "General Medicine"))[0]
 
-    prompt = f"""As a {specialist.lower()}, explain {condition} to a patient experiencing these symptoms: "{symptoms}".
+    prompt = f"""As a {specialist_title.lower()}, explain {condition} to a patient experiencing these symptoms: "{symptoms}". 
+    
     Structure the response into:
-    1. **Biological Process**
-    2. **Immediate Treatment**
-    3. **Long-term Care**
-    4. **Emergency Signs**
-    5. **Diet Plan**
+    1. **Biological Process**  
+    2. **Immediate Treatment**  
+    3. **Long-term Care**  
+    4. **Emergency Signs**  
+    5. **Diet Plan**  
     Use professional yet simple language.
     """
 
@@ -69,95 +79,94 @@ def generate_expert_analysis(condition, symptoms):
 def create_medical_report(symptoms):
     try:
         translated_symptoms = translate_text(symptoms, "en")
-        result = classifier(translated_symptoms, list(conditions.keys()), multi_label=False)
+        result = classifier(translated_symptoms, conditions, multi_label=False)
         diagnosis = result['labels'][0]  
-        specialist = conditions.get(diagnosis, "General Physician")
+        specialist, system = specialist_mapping.get(diagnosis, ("General Physician", "General Medicine"))
 
         expert_analysis = generate_expert_analysis(diagnosis, translated_symptoms)
 
-        full_report = f"""
-        ### 📋 Medical Report
-        **📝 Patient Symptoms:** {translated_symptoms}  
-        **🔍 Primary Diagnosis:** {diagnosis}  
-        **🫀 Consult:** {specialist}  
-
-        **🧑‍⚕️ Expert Analysis:**  
-        {expert_analysis}
-
-        **🩺 Key Questions for Doctor:**  
-        1️⃣ Is this condition acute or chronic?  
-        2️⃣ What medication options are suitable?  
-        3️⃣ What lifestyle changes help manage this condition?  
-        4️⃣ What warning signs require immediate attention?
-        """
+        full_report = (
+            f"**Medical Report**\n\n"
+            f"**Patient Symptoms:** {translated_symptoms}\n"
+            f"**Primary Diagnosis:** {diagnosis}\n"
+            f"**Affected System:** {system}\n"
+            f"**Consult:** {specialist}\n\n"
+            f"**Expert Analysis:**\n{expert_analysis}\n\n"
+        )
 
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", size=12)
         pdf.multi_cell(0, 10, full_report)
-        pdf.output("medical_report.pdf")
+        pdf_path = "medical_report.pdf"
+        pdf.output(pdf_path)
 
-        return full_report, "medical_report.pdf"
+        return full_report, pdf_path
 
     except Exception as e:
         return f"Error generating report: {str(e)}", None
 
-# Streamlit UI
+# Streamlit UI Enhancements
 st.set_page_config(page_title="MedExpert AI", layout="wide")
 
-# Custom CSS for Tailwind Styling
-st.markdown("""
+# Apply Tailwind CSS & Custom Styles
+st.markdown(
+    """
     <style>
-    body { background: linear-gradient(135deg, #1d3557, #457b9d); color: white; }
-    .sidebar .sidebar-content { background: #1d3557 !important; color: white; }
-    .stButton>button { background: #e63946; color: white; font-weight: bold; padding: 12px 20px; border-radius: 5px; }
-    .stButton>button:hover { background: #a02938; }
-    .report-box { background: white; color: black; padding: 20px; border-radius: 10px; }
-    .loader { animation: spin 1s linear infinite; border: 4px solid #ccc; border-top: 4px solid #e63946; border-radius: 50%; width: 50px; height: 50px; }
-    @keyframes spin { from { transform: rotate(0deg); } to { transform: rotate(360deg); } }
+        body {background: linear-gradient(to right, #ff7e5f, #feb47b);}
+        .sidebar .sidebar-content {background: #2C3E50;}
+        .sidebar .sidebar-content a {color: #ffffff !important; font-size: 18px;}
+        .stButton>button {background-color: #16A085; color: white; padding: 10px 20px; border-radius: 8px;}
+        .stButton>button:hover {background-color: #1ABC9C;}
+        .stMarkdown {font-size: 18px; color: #2C3E50;}
     </style>
-""", unsafe_allow_html=True)
+    """, 
+    unsafe_allow_html=True
+)
 
-# Sidebar
-st.sidebar.markdown("## 🔍 Navigation")
-page = st.sidebar.radio("", ["🏠 Home", "📄 Diagnosis", "📢 About"])
+# Sidebar UI
+with st.sidebar:
+    st.markdown("## 🔍 Navigation")
+    page = st.radio("", ["🏠 Home", "📄 Diagnosis", "📢 About"])
+    st.markdown("---")
+    st.markdown("**📌 Built by AI & Healthcare Experts**")
 
 # Main Page
-st.markdown("<h1 style='text-align: center;'>🩺 MedExpert AI</h1>", unsafe_allow_html=True)
-st.markdown("<p style='text-align: center; font-size: 18px;'>💡 AI-powered medical diagnosis & reports</p>", unsafe_allow_html=True)
+st.title("🩺 MedExpert: AI-Powered Clinical Analysis System")
+st.markdown("💡 *Get AI-powered medical analysis & professional reports instantly!*")
 
+# Home Page
 if page == "🏠 Home":
-    st.image("https://source.unsplash.com/800x400/?healthcare,medical", use_column_width=True)
-    st.markdown("""
-    - 🔹 AI-based clinical **symptom analysis**  
-    - 🔹 Generates **professional medical reports**  
-    - 🔹 Provides **expert doctor recommendations**  
-    - 🔹 Supports **multiple languages**
-    """)
-
+    st.image("https://source.unsplash.com/800x400/?healthcare,hospital", use_container_width=True)
+    st.write("### Welcome to MedExpert!")
+    st.write("🔹 AI-based clinical diagnosis system for analyzing symptoms.")
+    st.write("🔹 Generates professional **medical reports** instantly.")
+    st.write("🔹 Provides **expert doctor recommendations**.")
+    st.write("🔹 Supports **multiple languages** (Urdu, English, Arabic, etc.).")
+
+# Diagnosis Page
 elif page == "📄 Diagnosis":
-    st.markdown("<h2>📝 Describe Your Symptoms</h2>", unsafe_allow_html=True)
+    st.subheader("📝 Describe Your Symptoms")
     symptoms = st.text_area("Enter symptoms in any language", placeholder="e.g., Mujhay sar main dard hai")
 
     if st.button("🧑‍⚕️ Get Diagnosis"):
         if symptoms.strip():
-            with st.spinner("🔍 AI is analyzing..."):
+            with st.spinner("🔍 Analyzing Symptoms..."):
+                time.sleep(2)  # Simulate processing
                 report_text, pdf_file = create_medical_report(symptoms)
             
-            if report_text:
-                st.markdown('<div class="report-box">', unsafe_allow_html=True)
-                st.markdown(report_text, unsafe_allow_html=True)
-                st.markdown('</div>', unsafe_allow_html=True)
-                st.success("✅ Report Generated Successfully!")
-
+            if pdf_file:
+                st.subheader("📋 Medical Report")
+                st.write(report_text)
                 with open(pdf_file, "rb") as file:
                     st.download_button(label="📥 Download Report", data=file, file_name="medical_report.pdf", mime="application/pdf")
+            else:
+                st.error("❌ Report generation failed.")
         else:
             st.error("⚠ Please enter symptoms before proceeding.")
 
+# About Page
 elif page == "📢 About":
-    st.markdown("### ℹ️ About MedExpert")
-    st.markdown("AI-powered **medical diagnosis system** to provide expert guidance.")
+    st.write("### ℹ️ About MedExpert")
+    st.write("This AI-powered medical system helps users get professional diagnosis and expert medical recommendations.")
 
-st.markdown("---")
-st.markdown("<p style='text-align: center;'>👨‍⚕️ Built with ❤️ by AI & Healthcare Experts</p>", unsafe_allow_html=True)