Upload 2 files

.gitattributes

@@ -37,3 +37,4 @@ a[[:space:]]dog[[:space:]]with[[:space:]]a[[:space:]]butterf.png filter=lfs diff
 dog_butterfly.png filter=lfs diff=lfs merge=lfs -text
 dog_butterfly_flap.png filter=lfs diff=lfs merge=lfs -text
 butterfly_cute.gif filter=lfs diff=lfs merge=lfs -text
+chocolate_lab_butterfly_pic.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -9,9 +9,7 @@ import os
 from datasets import load_dataset, Dataset
 from huggingface_hub import login
 
-# Authenticate with your HF token stored as a secret
 login(token=os.getenv("sitetoken"))
-
 DATASET_NAME = "Mephesto1/Diabetic-tracker"
 
 # Load or create dataset
@@ -45,12 +43,7 @@ def load_last_inputs():
             sensor_site = ""
         if pod_site not in [s["site"] for s in sites]:
             pod_site = ""
-        return (
-            sensor_site,
-            last["sensor_date"],
-            pod_site,
-            last["pod_date"]
-        )
+        return (sensor_site, last["sensor_date"], pod_site, last["pod_date"])
     return "", "", "", ""
 
 def convert_date_format(date_input):
@@ -108,7 +101,6 @@ def generate_schedules(sensor_site, sensor_date, pod_site, pod_date):
 
     site_rotation = {}
 
-        # 🌟 Apply selected starting pod site for first change
     sensor_info_for_start = next(
         (s for s in reversed(sensor_schedule) if s["date"] <= pod_date),
         sensor_schedule[0]
@@ -120,12 +112,10 @@ def generate_schedules(sensor_site, sensor_date, pod_site, pod_date):
 
     for i in range(pod_num_changes):
         pod_change_date = pod_start_date + timedelta(days=i * pod_change_days)
-
         sensor_info = next(
             (s for s in reversed(sensor_schedule) if s["date"] <= pod_change_date.strftime("%Y-%m-%d")),
             sensor_schedule[0]
         )
-
         key = f"{sensor_info['site']}_{sensor_info['date']}"
         available_pod_sites = pod_sites_map.get(sensor_info["site"], ["Left Stomach", "Right Stomach"])
 
@@ -150,9 +140,7 @@ def generate_schedules(sensor_site, sensor_date, pod_site, pod_date):
         elif type_ == "pod":
             return "\n".join([f"Pod on {entry['date']} at {entry['pod_site']} — Sensor is on {entry['sensor_site']}" for entry in schedule])
 
-    formatted_sensor_schedule = format_schedule(sensor_schedule, "sensor")
-    formatted_pod_schedule = format_schedule(pod_schedule, "pod")
-    return formatted_sensor_schedule, formatted_pod_schedule
+    return format_schedule(sensor_schedule, "sensor"), format_schedule(pod_schedule, "pod")
 
 pod_sites_map = {
     "Left Arm": ["Left Leg", "Left Stomach", "Left Arm"],
@@ -161,46 +149,40 @@ pod_sites_map = {
     "Right Arm": ["Right Leg", "Right Stomach", "Right Arm"]
 }
 
-# 🦋 Load both image frames
-dog_static = Image.open("dog_butterfly.png")
-dog_flap = Image.open("dog_butterfly_flap.png")  # Nora’s fluttering buddy!
+# 🐶 Load image and gif
+display_image = Image.open("chocolate_lab_butterfly_pic.png")
+butterfly_gif = Image.open("butterfly_cute.gif")
 
-def flutter_effect(sensor_site, sensor_date, pod_site, pod_date):
-    sensor_sched, pod_sched = generate_schedules(sensor_site, sensor_date, pod_site, pod_date)
-    yield dog_flap, sensor_sched, pod_sched
-    time.sleep(0.4)
-    yield dog_static, sensor_sched, pod_sched
+def show_gif_after_save(sensor_site, sensor_date, pod_site, pod_date):
+    save_to_dataset(sensor_site, sensor_date, pod_site, pod_date)
+    return butterfly_gif
 
 # 🎛️ Build the interface
 with gr.Blocks() as demo:
-    gr.Markdown("### Hi Nora! Let’s update your next change. 🐶🦋")
+    gr.Markdown("### Hello, let's plan your new pod and sensor change. 🐶🦋")
 
     last_sensor_site, last_sensor_date, last_pod_site, last_pod_date = load_last_inputs()
     dropdown_sites = [site["site"] for site in sites]
 
+    gr.Image(value=display_image, show_label=False)
+
+    sensor_sched, pod_sched = generate_schedules(last_sensor_site, last_sensor_date, last_pod_site, last_pod_date)
+    sensor_output = gr.Textbox(label="Sensor Schedule", lines=10, value=sensor_sched, interactive=True)
+    pod_output = gr.Textbox(label="Pod Schedule", lines=10, value=pod_sched, interactive=True)
+
     sensor_site = gr.Dropdown(dropdown_sites, label="Sensor Site", value=last_sensor_site)
     sensor_date = Calendar(type="datetime", label="Sensor Change Date", value=last_sensor_date)
     pod_site = gr.Dropdown(dropdown_sites, label="Pod Site", value=last_pod_site)
     pod_date = Calendar(type="datetime", label="Pod Change Date", value=last_pod_date)
 
-    butterfly_display = gr.Image(value=dog_static, label="Butterfly Buddy", show_label=False)
-    
-    sensor_output = gr.Textbox(label="Sensor Schedule", lines=10, value="", interactive=True)
-    pod_output = gr.Textbox(label="Pod Schedule", lines=10, value="", interactive=True)
+    butterfly_display = gr.Image(show_label=False)
 
     paw_button = gr.Button("🐾 Press your paw here to save!")
-    paw_button.click(fn=save_to_dataset, inputs=[sensor_site, sensor_date, pod_site, pod_date], outputs=[])
-
-    for c in [sensor_site, sensor_date, pod_site, pod_date]:
-        c.change(fn=flutter_effect,
-                 inputs=[sensor_site, sensor_date, pod_site, pod_date],
-                 outputs=[butterfly_display, sensor_output, pod_output])
-
-    # Trigger image + schedules on load
-    sensor_sched, pod_sched = generate_schedules(last_sensor_site, last_sensor_date, last_pod_site, last_pod_date)
-    butterfly_display.value = dog_static
-    sensor_output.value = sensor_sched
-    pod_output.value = pod_sched
+    paw_button.click(
+        fn=show_gif_after_save,
+        inputs=[sensor_site, sensor_date, pod_site, pod_date],
+        outputs=butterfly_display
+    )
 
 if __name__ == "__main__":
     demo.launch()