Update app.py

app.py

@@ -10,30 +10,41 @@ with open("stroke_clf.pkl", "rb") as f:
 with open("stroke_reg.pkl", "rb") as f:
     reg_model = pickle.load(f)
 
-def predict_stroke(age, chest_pain, shortness_breath, irregular_heartbeat, fatigue_weakness, 
-                  dizziness, swelling, pain_neck_jaw, excessive_sweating, persistent_cough, 
-                  nausea_vomiting, high_bp, chest_discomfort, cold_hands_feet, snoring, anxiety):
+# Custom CSS for enhanced styling
+custom_css = """
+.gradio-container {
+    max-width: 1200px !important;
+    margin-left: auto !important;
+    margin-right: auto !important;
+}
+
+.container {
+    margin: 0 auto !important;
+    padding: 2rem !important;
+}
+
+.question-group {
+    border: 1px solid #e5e7eb !important;
+    border-radius: 8px !important;
+    padding: 1.5rem !important;
+    margin-bottom: 1.5rem !important;
+    background: white !important;
+}
+
+.footer {
+    text-align: center !important;
+    padding: 2rem !important;
+    background: #f8fafc !important;
+    border-top: 1px solid #e5e7eb !important;
+}
+"""
+
+def predict_stroke(*inputs):
     """
-    Prepares input features and returns stroke risk prediction using both models.
+    Enhanced prediction function with detailed output formatting
     """
-    features = [
-        1 if chest_pain else 0,
-        1 if shortness_breath else 0,
-        1 if irregular_heartbeat else 0,
-        1 if fatigue_weakness else 0,
-        1 if dizziness else 0,
-        1 if swelling else 0,
-        1 if pain_neck_jaw else 0,
-        1 if excessive_sweating else 0,
-        1 if persistent_cough else 0,
-        1 if nausea_vomiting else 0,
-        1 if high_bp else 0,
-        1 if chest_discomfort else 0,
-        1 if cold_hands_feet else 0,
-        1 if snoring else 0,
-        1 if anxiety else 0,
-        age
-    ]
+    age = inputs[0]
+    features = [1 if x == "Yes" else 0 for x in inputs[1:]] + [age]
     
     sample_input = np.array([features])
     
@@ -42,224 +53,235 @@ def predict_stroke(age, chest_pain, shortness_breath, irregular_heartbeat, fatig
     regression_result = reg_model.predict(sample_input)[0]
     risk_percentage = round(regression_result, 2)
     
-    # Simulate processing time for better UX
-    time.sleep(1)
+    # Simulate processing
+    time.sleep(1.5)
     
-    # Format the response based on risk level
+    # Enhanced results formatting
     if classification_result == 1:
+        severity = "High" if risk_percentage > 70 else "Moderate"
+        color = "red" if risk_percentage > 70 else "orange"
         result = f"""
-## 🚨 Risk Assessment Results
-
-**Status:** At Risk
-**Risk Level:** {risk_percentage}%
-
-### Important Next Steps:
-1. 👨‍⚕️ Schedule an immediate appointment with your healthcare provider
-2. 📊 Monitor your blood pressure regularly
-3. 💊 Take prescribed medications as directed
-4. 📝 Maintain a log of any symptoms you experience
-
-### Lifestyle Recommendations:
-- 🏃‍♂️ Start gentle exercise after consulting your doctor
-- 🥗 Follow a heart-healthy diet
-- 😴 Ensure 7-8 hours of quality sleep
-- 🚭 Quit smoking if applicable
-- 🧘‍♂️ Practice stress management techniques
+        <div style='background-color: #FEF2F2; padding: 20px; border-radius: 10px; border: 1px solid {color}'>
+            <h2 style='color: {color}; margin-bottom: 15px'>🚨 Risk Assessment Results: {severity} Risk</h2>
+            <div style='background-color: white; padding: 15px; border-radius: 8px; margin-bottom: 15px'>
+                <p style='font-size: 18px; margin-bottom: 10px'><strong>Risk Level:</strong> {risk_percentage}%</p>
+                <p style='color: {color}'><strong>Status:</strong> Immediate Attention Recommended</p>
+            </div>
+            
+            <h3 style='margin: 20px 0 10px 0'>📋 Recommended Actions:</h3>
+            <ol style='margin-left: 20px'>
+                <li>Schedule an urgent appointment with your healthcare provider</li>
+                <li>Begin daily blood pressure monitoring</li>
+                <li>Review and log all current medications</li>
+                <li>Start a symptom diary</li>
+            </ol>
 
-Remember: Early intervention is key to preventing stroke. Please consult a healthcare professional as soon as possible.
-"""
+            <h3 style='margin: 20px 0 10px 0'>🌟 Immediate Lifestyle Changes:</h3>
+            <ul style='margin-left: 20px'>
+                <li>Reduce sodium intake</li>
+                <li>Increase water consumption</li>
+                <li>Practice stress-reduction techniques</li>
+                <li>Ensure 7-8 hours of quality sleep</li>
+            </ul>
+        </div>
+        """
     else:
         result = f"""
-## ✅ Risk Assessment Results
-
-**Status:** Low Risk
-**Risk Level:** {risk_percentage}%
-
-### Maintaining Your Health:
-1. 🏃‍♂️ Continue regular exercise
-2. 🥗 Maintain a balanced diet
-3. 🩺 Schedule regular check-ups
-4. 💧 Stay hydrated
-5. 😌 Manage stress levels
-
-### Preventive Measures:
-- 🌟 Regular physical activity (150 minutes/week)
-- 🍎 Five servings of fruits and vegetables daily
-- 💤 Consistent sleep schedule
-- 🧘‍♂️ Regular relaxation practices
-- 📊 Annual health screenings
+        <div style='background-color: #F0FDF4; padding: 20px; border-radius: 10px; border: 1px solid #22C55E'>
+            <h2 style='color: #22C55E; margin-bottom: 15px'>✅ Risk Assessment Results: Low Risk</h2>
+            <div style='background-color: white; padding: 15px; border-radius: 8px; margin-bottom: 15px'>
+                <p style='font-size: 18px; margin-bottom: 10px'><strong>Risk Level:</strong> {risk_percentage}%</p>
+                <p style='color: #22C55E'><strong>Status:</strong> Healthy Range</p>
+            </div>
+            
+            <h3 style='margin: 20px 0 10px 0'>🌟 Maintaining Your Health:</h3>
+            <ul style='margin-left: 20px'>
+                <li>Continue regular exercise routine</li>
+                <li>Maintain balanced nutrition</li>
+                <li>Schedule regular check-ups</li>
+                <li>Practice stress management</li>
+            </ul>
 
-Keep up the good work in maintaining your health!
-"""
+            <h3 style='margin: 20px 0 10px 0'>💪 Preventive Measures:</h3>
+            <ul style='margin-left: 20px'>
+                <li>Annual health screenings</li>
+                <li>Regular blood pressure checks</li>
+                <li>Adequate sleep and hydration</li>
+                <li>Active lifestyle maintenance</li>
+            </ul>
+        </div>
+        """
     
     return result
 
-# Create the Gradio interface with a more interactive layout
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="purple")) as demo:
-    # Header
-    gr.Markdown(
-    """
-    # **Welcome to EarlyMed Stroke Risk Prediction Tool! 🏥🧠**
-    
-    Welcome to **EarlyMed**—an initiative by our team at **VIT-AP University**, dedicated to empowering you with early health insights. 
-    **Leveraging AI for early detection**, our mission is simple: ✨ **"Early Detection, Smarter Decision."**
+# Create the enhanced Gradio interface
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="purple"),
+    title="EarlyMed: AI-Powered Stroke Risk Assessment",
+    css=custom_css
+) as demo:
     
-    As part of our commitment to preventive healthcare, we have developed the **EarlyMed Stroke Risk Prediction Tool** 
-    to help you assess your stroke risk before consulting a doctor. This tool is designed to provide **timely insights** 
-    and encourage proactive health decisions.
+    # Header Section
+    gr.HTML("""
+        <div style="text-align: center; padding: 2rem; background: linear-gradient(to right, #E8F0FF, #F0F7FF);">
+            <h1 style="color: #1E40AF; margin-bottom: 1rem;">🌟 EarlyMed: AI-Powered Stroke Risk Assessment 🧠</h1>
+            <p style="color: #4B5563; font-size: 1.1rem; max-width: 800px; margin: 0 auto;">
+                Welcome to <strong>EarlyMed</strong>—a VIT-AP University initiative using advanced AI to help you understand your stroke risk factors.
+            </p>
+        </div>
     """)
     
     with gr.Row():
         with gr.Column():
             # Basic Information
-            gr.Markdown("### 👤 Basic Information")
-            age = gr.Slider(label="What is your age?", minimum=18, maximum=100, step=1, value=50)
-            
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 👤 Your Profile")
+                age = gr.Slider(
+                    label="How old are you?",
+                    minimum=18,
+                    maximum=100,
+                    step=1,
+                    value=50,
+                    info="Move the slider to your age"
+                )
+
             # Primary Symptoms
-            gr.Markdown("### 🫀 Primary Symptoms")
-            chest_pain = gr.Radio(
-                label="First of all, are you experiencing any chest pain?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            shortness_breath = gr.Radio(
-                label="Do you feel short of breath during normal activities?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            irregular_heartbeat = gr.Radio(
-                label="Have you noticed irregular heartbeats?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 💗 Heart & Chest Symptoms")
+                chest_pain = gr.Radio(
+                    label="Do you experience any discomfort or pain in your chest?",
+                    choices=["No", "Yes"],
+                    value="No",
+                    info="This might feel like pressure, squeezing, or general discomfort"
+                )
+                shortness_breath = gr.Radio(
+                    label="Do you find yourself getting out of breath easily?",
+                    choices=["No", "Yes"],
+                    value="No",
+                    info="For example, during light activities like climbing stairs"
+                )
+                irregular_heartbeat = gr.Radio(
+                    label="Have you noticed your heart beating irregularly?",
+                    choices=["No", "Yes"],
+                    value="No",
+                    info="This might feel like skipped beats or fluttering"
+                )
+
         with gr.Column():
-            # Secondary Symptoms
-            gr.Markdown("### 🤒 Additional Symptoms")
-            fatigue_weakness = gr.Radio(
-                label="Now I am gonna check some additional features. Do you feel unusual fatigue or weakness?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            dizziness = gr.Radio(
-                label="Do you experience dizziness?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            swelling = gr.Radio(
-                label="Have you noticed swelling in your extremities?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            
+            # Daily Symptoms
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 😮‍💨 Daily Experiences")
+                fatigue_weakness = gr.Radio(
+                    label="Do you often feel unusually tired or weak?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                dizziness = gr.Radio(
+                    label="Do you experience dizziness or light-headedness?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                swelling = gr.Radio(
+                    label="Have you noticed swelling in your legs or ankles?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+
     with gr.Row():
         with gr.Column():
-            # Additional Health Indicators
-            gr.Markdown("### 🏥 Health Indicators")
-            pain_neck_jaw = gr.Radio(
-                label="Do you have pain in neck, jaw, or shoulders?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            excessive_sweating = gr.Radio(
-                label="Do you experience excessive sweating?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            persistent_cough = gr.Radio(
-                label="Do you have a persistent cough?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            
+            # Additional Symptoms
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 🤒 Additional Signs")
+                pain_neck_jaw = gr.Radio(
+                    label="Do you experience pain in your neck, jaw, or shoulders?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                excessive_sweating = gr.Radio(
+                    label="Do you sweat more than usual, even when not active?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                persistent_cough = gr.Radio(
+                    label="Have you developed a cough that won't go away?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+
         with gr.Column():
-            # Risk Factors
-            gr.Markdown("### ⚠️ Risk Factors")
-            nausea_vomiting = gr.Radio(
-                label="Do you experience nausea or vomiting?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            high_bp = gr.Radio(
-                label="Have you been diagnosed with high blood pressure?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            chest_discomfort = gr.Radio(
-                label="Do you feel chest discomfort during activities?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            
+            # Health Conditions
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 🏥 Health Conditions")
+                nausea_vomiting = gr.Radio(
+                    label="Do you feel nauseated or sick to your stomach?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                high_bp = gr.Radio(
+                    label="Has a doctor ever told you that you have high blood pressure?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                chest_discomfort = gr.Radio(
+                    label="Do you feel discomfort in your chest when you're active?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+
     with gr.Row():
         with gr.Column():
             # Lifestyle Factors
-            gr.Markdown("### 🌟 Lifestyle Factors")
-            cold_hands_feet = gr.Radio(
-                label="Do your hands or feet often feel cold?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            snoring = gr.Radio(
-                label="Do you snore or have sleep apnea?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-            anxiety = gr.Radio(
-                label="Do you experience frequent anxiety?",
-                choices=["No", "Yes"],
-                value="No"
-            )
-    
+            with gr.Group(elem_classes="question-group"):
+                gr.Markdown("### 🌟 Lifestyle & Well-being")
+                cold_hands_feet = gr.Radio(
+                    label="Do your hands or feet often feel unusually cold?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                snoring = gr.Radio(
+                    label="Has anyone told you that you snore loudly or stop breathing during sleep?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+                anxiety = gr.Radio(
+                    label="Do you often feel anxious or experience a sense of doom?",
+                    choices=["No", "Yes"],
+                    value="No"
+                )
+
     # Submit Button
-    submit_btn = gr.Button("Analyze My Risk", variant="primary")
+    gr.Markdown("""<div style='text-align: center; margin: 2rem 0;'>""")
+    submit_btn = gr.Button(
+        "Analyze My Risk Factors",
+        variant="primary",
+        size="lg"
+    )
     
     # Output
-    output = gr.Markdown(label="Risk Assessment Results")
-    
-    # Process inputs
-    def process_inputs(*args):
-        # Convert "Yes"/"No" to boolean
-        bool_args = [arg == "Yes" for arg in args[1:]]
-        return predict_stroke(args[0], *bool_args)
+    output = gr.HTML(label="Assessment Results")
     
+    # Wire up the prediction
     submit_btn.click(
-        process_inputs,
-        inputs=[age, chest_pain, shortness_breath, irregular_heartbeat, 
+        predict_stroke,
+        inputs=[age, chest_pain, shortness_breath, irregular_heartbeat,
                 fatigue_weakness, dizziness, swelling, pain_neck_jaw,
                 excessive_sweating, persistent_cough, nausea_vomiting,
                 high_bp, chest_discomfort, cold_hands_feet, snoring, anxiety],
         outputs=output
     )
     
-    # Information Section
-    gr.Markdown(
-    """
-    ---
-    ### **How Does This Work? 🤖**
-    Our AI model analyzes key health factors—such as **age, blood pressure, lifestyle habits, and medical history**—to predict your stroke risk with high accuracy. The tool uses **two approaches**:
-    
-    🔹 **Classification Model:** Determines whether you are at risk or not.
-    🔹 **Regression Model:** Estimates your stroke risk percentage.
-    
-    ### **What to Do If Your Stroke Risk is High? 🚨**
-    If the tool predicts a high risk of stroke, don't panic! Here's what you should do:
-    
-    ✅ **Consult a Doctor:** Share your results with a healthcare professional for expert advice.
-    ✅ **Monitor Your Health:** Keep track of your **blood pressure, cholesterol levels, and overall fitness.**
-    ✅ **Adopt a Healthy Lifestyle:** Regular exercise, a balanced diet, and quitting smoking can significantly reduce stroke risks.
-    ✅ **Manage Chronic Conditions:** If you have **diabetes or hypertension**, ensure they are well-controlled.
-    
-    💡 **Important:** This tool is meant for **early awareness** and is **not a substitute for medical diagnosis**. Always consult a doctor for professional assessment and guidance.
-    
-    Stay informed, stay healthy, and take control of your well-being! 💙💪
-    
-    ---
-    
-    *© 2025 EarlyMed - VIT-AP University*
-    
-    **Disclaimer:** This tool is for informational purposes only and should not be used as a substitute for professional medical advice, diagnosis, or treatment.
-    """
-    )
+    # Footer
+    gr.HTML("""
+        <div style="text-align: center; margin-top: 2rem; padding: 2rem; background: linear-gradient(to right, #F8FAFC, #F0F7FF);">
+            <h3 style="color: #1E40AF; margin-bottom: 1rem;">💡 Important Note</h3>
+            <p style="color: #4B5563; max-width: 800px; margin: 0 auto 1rem auto;">
+                This tool is designed for educational purposes and early awareness only. It is not a substitute for professional medical diagnosis or advice.
+            </p>
+            <p style="color: #6B7280; font-size: 0.9rem;">
+                © 2025 EarlyMed - VIT-AP University | All Rights Reserved
+            </p>
+        </div>
+    """)
 
+# Launch the interface
 demo.launch()