Update app.py

app.py

@@ -2,92 +2,107 @@ import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 
-# Constants for Galvus (Vildagliptin) effects based on studies
-post_meal_reduction_mean = 63.6  # mg/dL
-post_meal_reduction_std = 74  # mg/dL
-fasting_glucose_reduction_mean = 27.7  # mg/dL
-fasting_glucose_reduction_std = 43.4  # mg/dL
-
-# Meal types and their carbs (in mL)
-meal_options = {
-    "Light Meal": 30,  # mL of fast carbs
-    "Standard Meal": 50,  # mL of fast carbs
-    "Heavy Meal": 70,  # mL of fast carbs
-    "Fast Carbs (Juice/Milk)": 120  # mL of fast carbs (for quick glucose rise)
-}
-
-# Function to calculate blood glucose prediction
-def predict_glucose(current_glucose, meal_type, meal_time, galvus_dose, exercise_duration, fast_carb_mL):
-    # Convert meal time to hours and minutes
-    if "min ago" in meal_time:
-        minutes_ago = int(meal_time.replace(" min ago", ""))
-        meal_time_hours = minutes_ago // 60
-        meal_time_minutes = minutes_ago % 60
-    else:
-        # Default to 0 if no valid input
-        meal_time_hours, meal_time_minutes = 0, 0
+# Function to simulate blood glucose changes over time
+def predict_glucose(current_glucose, meal_type, meal_time_hours, meal_time_minutes, galvus_dose, galvus_time_hours, galvus_time_minutes, exercise_duration, fast_carbs_ml, prediction_time=3):
+    # Constants for glucose reduction effects
+    post_meal_reduction = 63.6  # mg/dL (avg reduction for Vildagliptin in 2 hours)
+    fasting_reduction = 27.7  # mg/dL (avg reduction over 6-12 hours)
     
-    # Calculate predicted glucose
-    # 1. Post-Meal Prediction (after 2 hours)
-    if current_glucose > 100:  # Adjust if high glucose
-        post_meal_pred = current_glucose - post_meal_reduction_mean
+    # Adjust for fast carbs (milk, juice, etc.)
+    carb_effect = fast_carbs_ml * 1.5  # Approximate glucose rise per mL of fast carbs
+
+    # Convert meal and Galvus times to total minutes
+    meal_time_total = meal_time_hours * 60 + meal_time_minutes
+    galvus_time_total = galvus_time_hours * 60 + galvus_time_minutes
+
+    # Calculate blood glucose over time considering meal type, Galvus dose, and exercise
+    if meal_type == 'High-carb':
+        glucose_after_meal = current_glucose + carb_effect
+    elif meal_type == 'Protein-heavy':
+        glucose_after_meal = current_glucose + 20  # Small glucose increase due to protein
+    elif meal_type == 'Low-carb':
+        glucose_after_meal = current_glucose - 10  # Small reduction due to low-carb meal
+    elif meal_type == 'Fast Carbs (Juice/Milk)':
+        glucose_after_meal = current_glucose + carb_effect  # Explicitly handle fast carbs meal
     else:
-        post_meal_pred = current_glucose  # No significant effect if glucose is normal/low
+        glucose_after_meal = current_glucose  # Normal meal with moderate carbs
 
-    # 2. Fasting Prediction (after 6-12 hours)
-    fasting_pred = current_glucose - fasting_glucose_reduction_mean
-    
-    # 3. Fast Carb Effect (adding juice/milk carbs)
-    carb_effect = fast_carb_mL * 2  # Simplified assumption: 2 mg/dL rise per 10 mL of liquid carb
-    
-    # Adjusting for exercise (duration, in minutes)
-    exercise_effect = exercise_duration * 0.5  # Assuming each minute of exercise reduces glucose by 0.5 mg/dL
-    
-    # Final prediction outputs after 1 hour, 3 hours
-    glucose_1hr = post_meal_pred + carb_effect - exercise_effect
-    glucose_3hr = fasting_pred + carb_effect - exercise_effect
-    
-    # Create a graph showing blood glucose over time
-    times = [0, 1, 3]
-    glucose_levels = [current_glucose, glucose_1hr, glucose_3hr]
-    
-    plt.plot(times, glucose_levels, marker='o')
-    plt.xlabel('Time (hours)')
-    plt.ylabel('Blood Glucose (mg/dL)')
-    plt.title('Blood Glucose Prediction Over Time')
-    
-    # Save plot as an image file
-    plt.savefig("/tmp/blood_glucose_prediction.png")
+    # Simulate blood glucose levels over 1 hour and 3 hours
+    glucose_1hr = glucose_after_meal - post_meal_reduction + carb_effect * 0.5  # Adjust for carb effect
+    glucose_3hr = glucose_1hr - fasting_reduction
+
+    # Adjust the effects of Galvus based on its administration time (medication effect starts at galvus_time + 1-2 hours)
+    time_since_galvus = (meal_time_total - galvus_time_total) / 60  # Convert back to hours
+    if time_since_galvus >= 1:  # After 1 hour, the effects of Galvus start kicking in
+        glucose_3hr -= fasting_reduction  # Galvus effect after 3 hours
+
+    # Exercise effect on glucose (hypothetical value, may vary based on intensity)
+    glucose_3hr -= exercise_duration * 2  # Exercise reduces glucose by 2 mg/dL per minute
+
+    # Plotting the graph of glucose prediction over time
+    time_points = [0, 1, 3]  # Time: 0 hours, 1 hour, 3 hours
+    glucose_values = [current_glucose, glucose_1hr, glucose_3hr]
+
+    plt.plot(time_points, glucose_values, marker='o', color='b')
+    plt.title("Blood Glucose Prediction Over Time")
+    plt.xlabel("Time (Hours)")
+    plt.ylabel("Blood Glucose (mg/dL)")
+    plt.xticks([0, 1, 2, 3])
+    plt.grid(True)
+    plt.tight_layout()
+
+    # Save the graph as a file to show it in Gradio
+    plt.savefig('/tmp/blood_glucose_prediction.png')
     plt.close()
 
-    # Return results
-    return glucose_1hr, glucose_3hr, "/tmp/blood_glucose_prediction.png"
+    # Return glucose predictions and the image file path
+    return glucose_1hr, glucose_3hr, '/tmp/blood_glucose_prediction.png'
 
 
-# Gradio interface to input data
-def update_fast_carb_visibility(meal_type):
-    if meal_type == "Fast Carbs (Juice/Milk)":
-        return gr.Slider(minimum=0, maximum=300, default=120, label="Amount of Fast Carbs (mL)")
-    else:
-        return gr.Slider(minimum=0, maximum=0, default=0, label="Amount of Fast Carbs (mL)")
-
-
-iface = gr.Interface(
-    fn=predict_glucose,
-    inputs=[
-        gr.Number(label="Current Blood Glucose (mg/dL)"),
-        gr.Dropdown(choices=list(meal_options.keys()), label="Meal Type"),
-        gr.Textbox(label="Time of Last Meal (e.g., '15 min ago')"),
-        gr.Number(label="Galvus Dose (50 mg)"),
-        gr.Number(label="Exercise Duration (minutes)"),
-        gr.Slider(minimum=0, maximum=300, default=120, label="Amount of Fast Carbs (mL)"),
-    ],
-    outputs=[
-        gr.Textbox(label="Blood Glucose After 1 Hour (mg/dL)"),
-        gr.Textbox(label="Blood Glucose After 3 Hours (mg/dL)"),
-        gr.Image(type="file", label="Blood Glucose Prediction Graph")
-    ],
-    live=True
-)
+# Gradio Interface
+def build_interface():
+    with gr.Blocks() as iface:
+        gr.Markdown("# Blood Glucose Prediction Model (With Vildagliptin Effects)")
+
+        # Inputs for current glucose, meal info, medication dose, exercise, fast carbs, and Galvus time
+        with gr.Row():
+            current_glucose = gr.Number(label="Current Blood Glucose (mg/dL)", value=150)
+            meal_type = gr.Radio(choices=["Normal", "High-carb", "Protein-heavy", "Low-carb", "Fast Carbs (Juice/Milk)"], label="Meal Type", value="Normal")
+
+        with gr.Row():
+            meal_time_hours = gr.Number(label="Last Meal Time (Hours Ago)", value=2)
+            meal_time_minutes = gr.Number(label="Last Meal Time (Minutes Ago)", value=0)
+
+        with gr.Row():
+            galvus_dose = gr.Number(label="Galvus Dose (mg)", value=50)
+            galvus_time_hours = gr.Number(label="Galvus Time of Administration (Hours Ago)", value=0)
+            galvus_time_minutes = gr.Number(label="Galvus Time of Administration (Minutes Ago)", value=0)
+
+        with gr.Row():
+            exercise_duration = gr.Number(label="Exercise Duration (min)", value=0)
+            fast_carbs_ml = gr.Number(label="Fast Carbs (mL)", value=0)
+
+        # Output predictions and graph
+        glucose_1hr_output = gr.Textbox(label="Predicted Glucose Level in 1 Hour (mg/dL)")
+        glucose_3hr_output = gr.Textbox(label="Predicted Glucose Level in 3 Hours (mg/dL)")
+        glucose_graph = gr.Image(label="Blood Glucose Prediction Graph")
+
+        # Button to trigger prediction
+        predict_button = gr.Button("Predict Blood Glucose")
+
+        # Set button action
+        predict_button.click(
+            predict_glucose,
+            inputs=[
+                current_glucose, meal_type, meal_time_hours, meal_time_minutes, galvus_dose,
+                galvus_time_hours, galvus_time_minutes, exercise_duration, fast_carbs_ml
+            ],
+            outputs=[glucose_1hr_output, glucose_3hr_output, glucose_graph]
+        )
+
+    return iface
+
 
+# Build and launch the Gradio interface
+iface = build_interface()
 iface.launch()