Update app.py

app.py

@@ -1,26 +1,25 @@
-import streamlit as st
+import gradio as gr
 from transformers import pipeline
 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")
+# ✅ Load Groq 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
+# ✅ Use Google Translate for auto-detection & translation
 def translate_text(text, target_lang="en"):
     try:
         return GoogleTranslator(source="auto", target=target_lang).translate(text)
     except Exception as e:
         return f"Translation Error: {str(e)}"
 
-# Medical Conditions
+# ✅ Load Models
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+# ✅ Medical Conditions List
 conditions = [
     "Asthma", "COPD", "Pneumonia", "Tuberculosis", "COVID-19", "Bronchitis",
     "Heart Failure", "Hypertension", "Diabetes Type 1", "Diabetes Type 2", 
@@ -28,7 +27,7 @@ conditions = [
     "Chronic Kidney Disease", "UTI", "Osteoporosis", "Psoriasis", "Epilepsy"
 ]
 
-# Specialist Mapping
+# ✅ Specialist Mapping
 specialist_mapping = {
     "Asthma": ("Pulmonologist", "Respiratory System"),
     "COPD": ("Pulmonologist", "Respiratory System"),
@@ -51,11 +50,11 @@ specialist_mapping = {
     "Epilepsy": ("Neurologist", "Nervous System")
 }
 
-# Function to generate expert analysis
 def generate_expert_analysis(condition, symptoms):
+    """Generates expert medical analysis using Groq API"""
     specialist_title = specialist_mapping.get(condition, ("General Physician", "General Medicine"))[0]
 
-    prompt = f"""As a {specialist_title.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**  
@@ -63,7 +62,8 @@ def generate_expert_analysis(condition, symptoms):
     3. **Long-term Care**  
     4. **Emergency Signs**  
     5. **Diet Plan**  
-    Use professional yet simple language.
+
+    Use professional yet simple language. **No AI disclaimers or generic advice**.
     """
 
     response = groq_client.chat.completions.create(
@@ -75,10 +75,11 @@ def generate_expert_analysis(condition, symptoms):
 
     return response.choices[0].message.content
 
-# Function to generate medical report
 def create_medical_report(symptoms):
+    """Generates a complete medical report"""
     try:
         translated_symptoms = translate_text(symptoms, "en")
+
         result = classifier(translated_symptoms, conditions, multi_label=False)
         diagnosis = result['labels'][0]  
         specialist, system = specialist_mapping.get(diagnosis, ("General Physician", "General Medicine"))
@@ -92,120 +93,53 @@ def create_medical_report(symptoms):
             f"**Affected System:** {system}\n"
             f"**Consult:** {specialist}\n\n"
             f"**Expert Analysis:**\n{expert_analysis}\n\n"
+            "**Key Questions for Your Doctor:**\n"
+            "1. Is this condition acute or chronic?\n"
+            "2. What medication options are suitable?\n"
+            "3. What lifestyle changes help manage this condition?\n"
+            "4. What warning signs require immediate attention?\n"
         )
 
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", size=12)
         pdf.multi_cell(0, 10, full_report)
-        pdf_path = "medical_report.pdf"
-        pdf.output(pdf_path)
+        pdf.output("report.pdf")
 
-        return full_report, pdf_path
+        return full_report, "report.pdf"
 
     except Exception as e:
         return f"Error generating report: {str(e)}", None
 
-# Streamlit UI Enhancements
-st.set_page_config(page_title="MedExpert AI", layout="wide")
-
-# Apply Tailwind CSS & Custom Styles
-st.markdown("""
-    <style>
-        /* Background Gradient */
-        body {
-            background: linear-gradient(to right, #ff7e5f, #feb47b);
-            color: white;
-        }
-
-        /* Sidebar Design */
-        [data-testid="stSidebar"] {
-            background-color: #2C3E50 !important;
-            color: white !important;
-        }
-
-        [data-testid="stSidebar"] a {
-            color: white !important;
-            font-size: 18px !important;
-        }
-
-        /* Custom Buttons */
-        .stButton>button {
-            background-color: #16A085;
-            color: white;
-            padding: 10px 20px;
-            border-radius: 8px;
-            font-size: 16px;
-        }
-
-        .stButton>button:hover {
-            background-color: #1ABC9C;
-        }
-
-        /* Custom Input Box */
-        .stTextInput>div>div>input {
-            background-color: white !important;
-            border-radius: 10px !important;
-            border: 2px solid #16A085 !important;
-            padding: 8px !important;
-        }
-
-        /* Title Styling */
-        .stApp h1 {
-            text-align: center;
-            font-size: 36px;
-            color: #2C3E50;
-        }
-
-    </style>
-""", unsafe_allow_html=True)
-
-
-# Sidebar UI
-with st.sidebar:
-    st.markdown("## 🔍 **Navigation**", unsafe_allow_html=True)
-    page = st.radio("", ["🏠 Home", "📄 Diagnosis", "📢 About"])
-    st.markdown("---")
-    st.markdown("### 📌 **Built with AI & Healthcare Experts**")
-
-
-# Main Page
-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,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.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("🔍 **Analyzing Symptoms... Please wait...**"):
-            time.sleep(2)  # Simulate processing
-            report_text, pdf_file = create_medical_report(symptoms)
-
+# ✅ Create Sidebar with Navigation
+with gr.Blocks(css="body { background-color: #f5f7fa; }") as app:
+    with gr.Row():
+        gr.Markdown("<h1 style='text-align: center; color: #4A90E2;'>MedExpert AI</h1>")
+    
+    with gr.Tabs():
+        with gr.TabItem("🏠 Home"):
+            gr.Markdown("""
+                <h2 style="color: #4A90E2;">Welcome to MedExpert</h2>
+                <p style="font-size:18px;">An AI-powered medical assistant that provides expert analysis based on your symptoms.</p>
+                <p style="font-size:18px;">Simply describe your symptoms, and let AI generate a detailed report!</p>
+            """)
+
+        with gr.TabItem("📄 Medical Diagnosis"):
+            with gr.Row():
+                symptoms_input = gr.Textbox(label="Describe Your Symptoms", placeholder="e.g., Mujhay sar main dard hai", interactive=True)
+                report_btn = gr.Button("🔍 Generate Report", elem_id="generate-btn")
             
-            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")
-            elif:
-                st.error("❌ Report generation failed.")
-        elif:
-            st.error("⚠ Please enter symptoms before proceeding.")
-
-# About Page
-elif page == "📢 About":
-    st.write("### ℹ️ About MedExpert")
-    st.write("This AI-powered medical system helps users get professional diagnosis and expert medical recommendations.")
+            with gr.Row():
+                report_output = gr.Textbox(label="Complete Medical Report", interactive=False)
+                pdf_output = gr.File(label="Download PDF Report")
+
+            report_btn.click(create_medical_report, inputs=[symptoms_input], outputs=[report_output, pdf_output])
+
+        with gr.TabItem("ℹ️ About"):
+            gr.Markdown("""
+                <h2 style="color: #4A90E2;">About This App</h2>
+                <p style="font-size:18px;">MedExpert AI is an intelligent health assistant designed to provide expert insights into medical conditions.</p>
+            """)
 
+# ✅ Launch App
+app.launch()