Update app.py

app.py

@@ -1,52 +1,69 @@
 import gradio as gr
-from transformers import pipeline
-
-# Load an instruction-tuned model for text2text generation
-generator = pipeline(
-    "text2text-generation",
-    model="google/flan-t5-small",
-    device=0  # set to -1 if you don’t have a GPU
-)
+import re
+from datetime import datetime
 
 def format_daily_plan(notes):
     """
-    Takes a comma-separated list of notes/to-dos with optional times
-    and returns a structured plan: 
-      - time-stamped events
-      - tasks list
+    Parses comma-separated notes/to-dos for time-stamped events and untimed tasks.
     """
-    # Build the prompt
-    prompt = (
-        "You are a personal assistant. "
-        "Convert these daily notes into a schedule and a task list.\n\n"
-        f"Notes: {notes}\n\n"
-        "Output format:\n"
-        "Today's Plan:\n\n"
-        "HH:MM AM/PM: Event description\n\n"
-        "Tasks: comma-separated list of remaining to-dos"
+    entries = [n.strip() for n in notes.split(",") if n.strip()]
+    schedule = []
+    tasks = []
+
+    time_pattern = re.compile(
+        r"""(?xi)
+        (?P<desc>.*?)              # description (non-greedy)
+        \s+at\s+                   # literal ' at '
+        (?P<hour>\d{1,2})          # hour
+        (?::(?P<minute>\d{2}))?    # optional :MM
+        \s*(?P<ampm>am|pm)?        # optional AM/PM
+        $
+        """
     )
-    
-    # Generate the formatted plan
-    output = generator(
-        prompt,
-        max_length=200,
-        do_sample=False
-    )[0]["generated_text"]
-    
-    return output
-
-# Gradio interface
+
+    for entry in entries:
+        m = time_pattern.match(entry)
+        if m:
+            desc = m.group("desc").strip().capitalize()
+            hour = int(m.group("hour"))
+            minute = int(m.group("minute") or 0)
+            ampm = m.group("ampm")
+            # normalize to 24h then back to formatted string
+            if ampm:
+                if ampm.lower() == "pm" and hour != 12:
+                    hour += 12
+                if ampm.lower() == "am" and hour == 12:
+                    hour = 0
+            time_str = datetime.strptime(f"{hour:02d}:{minute:02d}", "%H:%M").strftime("%I:%M %p")
+            schedule.append((time_str, desc))
+        else:
+            tasks.append(entry.capitalize())
+
+    # sort schedule by time
+    schedule.sort(key=lambda x: datetime.strptime(x[0], "%I:%M %p"))
+
+    # build output strings
+    plan_lines = ["Today's Plan:\n"]
+    for t, d in schedule:
+        plan_lines.append(f"{t}: {d}")
+    if tasks:
+        plan_lines.append("\nTasks: " + ", ".join(tasks))
+
+    return "\n".join(plan_lines)
+
+
+# Gradio UI
 iface = gr.Interface(
     fn=format_daily_plan,
     inputs=gr.Textbox(
-        lines=3,
-        placeholder="e.g. meeting at 9, lunch w/ Sarah, clean room, finish math hw, check bank acc, get groceries"
+        lines=2,
+        placeholder="e.g. meeting at 9, lunch w/ Sarah at 12pm, clean room, finish math hw"
     ),
     outputs=gr.Textbox(label="Today's Plan"),
     title="🗒️ Daily Notes → Structured Plan",
     description=(
-        "Paste your comma-separated notes or to-do list; "
-        "the model will output a time-based schedule and a tasks list."
+        "Quickly turn your free-form notes into a time-based schedule and task list—"
+        "no AI model needed, runs in milliseconds."
     ),
     theme="default"
 )