Update app.py

app.py

@@ -1,144 +1,95 @@
 import gradio as gr
+import datetime
 import numpy as np
-from datetime import datetime
+import matplotlib.pyplot as plt
 
-# Function to calculate time difference in hours
-def time_difference(last_time):
-    now = datetime.now()
-    try:
-        # Parse the input time in 12-hour format with AM/PM
-        last_time = datetime.strptime(last_time, "%I:%M %p")  # %I for 12-hour format, %p for AM/PM
-        diff = (now - last_time).total_seconds() / 3600  # Convert to hours
-        return max(0, diff)  # Ensure non-negative
-    except ValueError:
-        return None  # Invalid input
-
-# Blood sugar prediction function
-def predict_glucose(current_glucose, last_meal, meal_time, last_med, med_time, exercise, fast_carb_amount_ml, liquid_type):
-    # Convert time inputs to hours elapsed
-    meal_hours = time_difference(meal_time)
-    med_hours = time_difference(med_time)
-
-    # Risk scoring system
-    risk_score_now = 0
-    risk_score_1hr = 0
-    risk_score_3hr = 0  # New 3-hour risk score
-
-    # Meal impact
-    if last_meal == "High Carb":
-        risk_score_now += 2
-        risk_score_1hr += 1  
-        risk_score_3hr -= 1  # Carbs wear off after 3 hours
-    elif last_meal == "Protein/Fat":
-        risk_score_now -= 1
-        risk_score_1hr -= 1
-        risk_score_3hr -= 1
-    elif last_meal == "Skipped":
-        risk_score_now -= 2  
-        risk_score_1hr -= 2  
-        risk_score_3hr -= 3  # Fasting for too long may cause hypoglycemia
-    elif last_meal == "Fast Carb (Juice, Glucose Tablet, etc.)":
-        # Calculate grams of fast carbs based on liquid type and amount in mL
-        if liquid_type == "Milk":
-            fast_carb_amount_grams = fast_carb_amount_ml * 1.03  # Milk density ~1.03g/mL
-        else:
-            fast_carb_amount_grams = fast_carb_amount_ml  # For other liquids like juice, 1g/mL
-
-        fast_carb_effect = fast_carb_amount_grams * 0.5  # Example: each unit of fast carb raises blood sugar by 5-10 mg/dL
-        risk_score_now += fast_carb_effect
-        risk_score_1hr += fast_carb_effect * 0.5  # Less effect after 1 hour
-        risk_score_3hr -= fast_carb_effect * 0.5  # Glucose may drop after fast carbs wear off
-
-    # Medication impact
-    if last_med in ["Insulin", "Sulfonylurea"] and med_hours < 5:
-        risk_score_now -= 3  
-        risk_score_1hr -= 2  
-        risk_score_3hr -= 3  # Long-acting insulin may cause delayed hypoglycemia
-    elif last_med == "Metformin":
-        risk_score_now -= 1  
-        risk_score_1hr -= 1
-        risk_score_3hr -= 1
-
-    # Exercise impact
-    if exercise > 30:
-        risk_score_now -= 2  
-        risk_score_1hr -= 1  
-        risk_score_3hr -= 2  # Delayed exercise-related hypoglycemia
-
-    # Time-based adjustments
-    if meal_hours > 4:
-        risk_score_now -= 1  
-        risk_score_1hr -= 2
-        risk_score_3hr -= 3
-    if med_hours > 8:
-        risk_score_now += 1  
-        risk_score_1hr += 2
-        risk_score_3hr += 3
-
-    # Predict glucose levels
-    predicted_glucose_now = current_glucose + (risk_score_now * 10)
-    predicted_glucose_1hr = predicted_glucose_now + (risk_score_1hr * 10)
-    predicted_glucose_3hr = predicted_glucose_1hr + (risk_score_3hr * 10)
-
-    # Keep within realistic limits
-    predicted_glucose_now = max(40, min(300, predicted_glucose_now))
-    predicted_glucose_1hr = max(40, min(300, predicted_glucose_1hr))
-    predicted_glucose_3hr = max(40, min(300, predicted_glucose_3hr))
-
-    # Determine risk level
-    def glucose_status(glucose):
-        if glucose < 70:
-            return "⚠️ Hypoglycemia Risk"
-        elif glucose > 180:
-            return "⚠️ Hyperglycemia Risk"
-        else:
-            return "✅ Normal Range"
-
-    status_now = glucose_status(predicted_glucose_now)
-    status_1hr = glucose_status(predicted_glucose_1hr)
-    status_3hr = glucose_status(predicted_glucose_3hr)
-
-    return (f"**Now:** {predicted_glucose_now:.1f} mg/dL → {status_now}\n"
-            f"**In 1 Hour:** {predicted_glucose_1hr:.1f} mg/dL → {status_1hr}\n"
-            f"**In 3 Hours:** {predicted_glucose_3hr:.1f} mg/dL → {status_3hr}")
-
-# Define Gradio interface
-fast_carb_amount_input = gr.Number(label="Amount of Fast Carbs (mL)", value=0)
-
-# Create the interface components
-meal_input = gr.Dropdown(["Fast Carb (Juice, Glucose Tablet, etc.)", "High Carb", "Balanced", "Protein/Fat", "Skipped"], label="Last Meal Type")
-meal_time_input = gr.Textbox(label="Time of Last Meal (hh:mm AM/PM)", placeholder="e.g. 1:30 PM")
-med_input = gr.Dropdown(["None", "Metformin", "Insulin", "Sulfonylurea"], label="Last Medication Taken")
-med_time_input = gr.Textbox(label="Time of Last Medication (hh:mm AM/PM)", placeholder="e.g. 8:00 AM")
-exercise_input = gr.Number(label="Exercise in Last 2 Hours (minutes)", value=0)
-liquid_type_input = gr.Dropdown(["Milk", "Juice"], label="Liquid Type for Fast Carb")
-
-# Function to show/hide the fast carb amount input field
-def update_fast_carb_visibility(last_meal):
-    return gr.update(visible=(last_meal == "Fast Carb (Juice, Glucose Tablet, etc.)"))
-
-# Define Gradio Interface in a Blocks context
-with gr.Blocks() as iface:
-    current_glucose_input = gr.Number(label="Current Blood Glucose (mg/dL)", value=100)
-    meal_input = gr.Dropdown(["Fast Carb (Juice, Glucose Tablet, etc.)", "High Carb", "Balanced", "Protein/Fat", "Skipped"], label="Last Meal Type")
-    meal_time_input = gr.Textbox(label="Time of Last Meal (hh:mm AM/PM)", placeholder="e.g. 1:30 PM")
-    med_input = gr.Dropdown(["None", "Metformin", "Insulin", "Sulfonylurea"], label="Last Medication Taken")
-    med_time_input = gr.Textbox(label="Time of Last Medication (hh:mm AM/PM)", placeholder="e.g. 8:00 AM")
-    exercise_input = gr.Number(label="Exercise in Last 2 Hours (minutes)", value=0)
-    fast_carb_amount_input = gr.Number(label="Amount of Fast Carbs (mL)", value=0)
-    liquid_type_input = gr.Dropdown(["Milk", "Juice"], label="Liquid Type for Fast Carb")
+# Function to calculate predicted blood glucose based on inputs
+def predict_glucose(blood_glucose_now, last_meal_time, meal_type, carb_amount, medication_taken, exercise_done):
+    # Convert input time (12-hour format) to 24-hour for calculation
+    last_meal_time = datetime.datetime.strptime(last_meal_time, "%I:%M %p").time()
+    
+    # Assuming standard carb impact for milk (fast carbs)
+    fast_carb_effect = 30  # Approximate rise in glucose from 120 ml of milk
+    slow_carb_effect = 10  # Approximate rise for slow carbs
+    
+    # Medication effects (Metformin)
+    metformin_effect = -20  # Approximate glucose decrease over time after Metformin intake (hours)
+    
+    # Estimate the current blood glucose change based on meal type
+    if meal_type == "Milk (Fast Carb)":
+        glucose_change = fast_carb_effect
+    else:
+        glucose_change = slow_carb_effect
+    
+    # Time-based adjustments for Metformin action
+    if medication_taken:
+        # Assuming Metformin starts working after 2-4 hours (gradually lowering glucose)
+        time_since_medication = (datetime.datetime.now() - datetime.datetime.combine(datetime.date.today(), last_meal_time)).seconds / 3600
+        if time_since_medication >= 2:
+            glucose_change += metformin_effect
+    
+    # Adjust for exercise
+    if exercise_done:
+        glucose_change -= 10  # Decrease glucose due to physical activity
+    
+    # Predict future glucose levels (after 1 hour and 3 hours)
+    glucose_now = blood_glucose_now + glucose_change
+    glucose_1hr = glucose_now + glucose_change * 0.5  # Prediction after 1 hour
+    glucose_3hr = glucose_now + glucose_change * 0.3  # Prediction after 3 hours
+    
+    # Determine risk
+    if glucose_now > 180:
+        risk_now = "⚠️ Hyperglycemia Risk"
+    elif glucose_now < 70:
+        risk_now = "⚠️ Hypoglycemia Risk"
+    else:
+        risk_now = "Normal"
+        
+    if glucose_1hr > 180:
+        risk_1hr = "⚠️ Hyperglycemia Risk"
+    elif glucose_1hr < 70:
+        risk_1hr = "⚠️ Hypoglycemia Risk"
+    else:
+        risk_1hr = "Normal"
+        
+    if glucose_3hr > 180:
+        risk_3hr = "⚠️ Hyperglycemia Risk"
+    elif glucose_3hr < 70:
+        risk_3hr = "⚠️ Hypoglycemia Risk"
+    else:
+        risk_3hr = "Normal"
+    
+    # Blood glucose trajectory (graph generation)
+    time_points = [0, 1, 3]  # Time in hours: Now, 1 hour, and 3 hours
+    glucose_values = [glucose_now, glucose_1hr, glucose_3hr]
+    
+    plt.figure(figsize=(8, 6))
+    plt.plot(time_points, glucose_values, marker='o', color='b', label="Predicted Glucose")
+    plt.title('Blood Glucose Trajectory Over Time')
+    plt.xlabel('Time (hours)')
+    plt.ylabel('Blood Glucose (mg/dL)')
+    plt.grid(True)
+    plt.axhline(y=180, color='r', linestyle='--', label="Hyperglycemia Threshold")
+    plt.axhline(y=70, color='g', linestyle='--', label="Hypoglycemia Threshold")
+    plt.legend(loc="best")
     
-    # Event to show/hide the fast carb input field
-    meal_input.change(fn=update_fast_carb_visibility, inputs=meal_input, outputs=fast_carb_amount_input)
+    # Save the graph as an image
+    plt_file_path = "/tmp/glucose_trajectory.png"
+    plt.savefig(plt_file_path)
     
-    output = gr.Textbox()
+    # Return the predicted values and trajectory image
+    result_text = f"Now: {glucose_now:.1f} mg/dL → {risk_now}\nIn 1 Hour: {glucose_1hr:.1f} mg/dL → {risk_1hr}\nIn 3 Hours: {glucose_3hr:.1f} mg/dL → {risk_3hr}"
     
-    # Call the prediction function on form submission
-    gr.Button("Predict").click(
-        fn=predict_glucose,
-        inputs=[current_glucose_input, meal_input, meal_time_input, med_input, med_time_input, exercise_input, fast_carb_amount_input, liquid_type_input],
-        outputs=output
-    )
+    return result_text, plt_file_path
+
+# Gradio interface
+iface = gr.Interface(fn=predict_glucose, 
+                     inputs=[gr.Number(label="Current Blood Glucose (mg/dL)"), 
+                             gr.Textbox(label="Last Meal Time (e.g., 12:00 AM)"),
+                             gr.Radio(label="Meal Type", choices=["Milk (Fast Carb)", "Slow Carbs"]),
+                             gr.Number(label="Carb Amount (grams)"),
+                             gr.Checkbox(label="Medication Taken (Metformin)"),
+                             gr.Checkbox(label="Exercise Done")],
+                     outputs=[gr.Textbox(), gr.Image(type="file")])
 
-# Launch the interface
 iface.launch()