Create app.py

app.py

@@ -0,0 +1,160 @@
+import streamlit as st
+from transformers import pipeline
+from fpdf import FPDF
+import os
+from groq import Groq  
+from deep_translator import GoogleTranslator  
+
+# Load API Key
+groq_api_key = os.getenv("groq_api_key")
+groq_client = Groq(api_key=groq_api_key)  
+
+# Translation Function
+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)}"
+
+# Load Model
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+# 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_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}".
+    
+    Structure the response into:
+    1. **Biological Process**  
+    2. **Immediate Treatment**  
+    3. **Long-term Care**  
+    4. **Emergency Signs**  
+    5. **Diet Plan**  
+
+    Use professional yet simple language. **No AI disclaimers or generic advice**.
+    """
+
+    response = groq_client.chat.completions.create(
+        messages=[{"role": "user", "content": prompt}],
+        model="mixtral-8x7b-32768",
+        temperature=0.5,
+        max_tokens=1024
+    )
+
+    return response.choices[0].message.content
+
+# Function to generate medical report
+def create_medical_report(symptoms):
+    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"))
+
+        expert_analysis = generate_expert_analysis(diagnosis, translated_symptoms)
+
+        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"
+            "**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.output("medical_report.pdf")
+
+        return full_report, "medical_report.pdf"
+
+    except Exception as e:
+        return f"Error generating report: {str(e)}", None
+
+# Streamlit UI
+st.set_page_config(page_title="MedExpert AI", layout="wide")
+
+# Sidebar
+st.sidebar.title("🔍 Navigation")
+page = st.sidebar.radio("Go to", ["🏠 Home", "📄 Diagnosis", "📢 About"])
+
+# 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_column_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..."):
+                report_text, pdf_file = create_medical_report(symptoms)
+            
+            if report_text:
+                st.subheader("📋 Medical Report")
+                st.write(report_text)
+                st.success("✅ Report Generated Successfully!")
+                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("⚠ 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.")
+
+# Footer
+st.markdown("---")
+st.write("👨‍⚕️ **Built with ❤️ by AI & Healthcare Experts**")
+