update app.py

app.py

@@ -17,6 +17,10 @@ index = faiss.read_index(index_path)
 model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 recent = deque(maxlen=120)
 
+def render(a):
+    ctx = ", ".join(a["context"])
+    return f"**{a['title']}**  \n{a['instruction']}  \nGoal: {a['goal']} • Duration: {a['duration_min']} min • Energy: {a['energy']} • Context: {ctx}"
+
 def pick_unique(idxs, goal, energy, max_minutes, ignore_goal=False, ignore_energy=False, ignore_minutes=False, need=3, used=None):
     if used is None:
         used = set()
@@ -38,7 +42,10 @@ def pick_unique(idxs, goal, energy, max_minutes, ignore_goal=False, ignore_energ
     return out, used
 
 def fill_random(need, used, goal, energy, max_minutes):
-    pool = [a for a in actions if a["title"] not in used and a["title"] not in recent and (not goal or a["goal"]==goal) and (not energy or a["energy"]==energy) and (not max_minutes or a["duration_min"]<=int(max_minutes))]
+    pool = [a for a in actions if a["title"] not in used and a["title"] not in recent
+            and (not goal or a["goal"] == goal)
+            and (not energy or a["energy"] == energy)
+            and (not max_minutes or a["duration_min"] <= int(max_minutes))]
     if len(pool) < need:
         pool = [a for a in actions if a["title"] not in used and a["title"] not in recent]
     random.shuffle(pool)
@@ -64,15 +71,42 @@ def search(query, goal, energy, max_minutes):
             res += fill_random(3-len(res), used, goal, energy, max_minutes)
 
         recent.extend([a["title"] for a in res[:3]])
-        out = []
-        for a in res[:3]:
-            out.append(f"**{a['title']}**  \n{a['instruction']}  \nGoal: {a['goal']} • Duration: {a['duration_min']} min • Energy: {a['energy']} • Context: {', '.join(a['context'])}")
-        return "\n\n---\n\n".join(out)
+        return "\n\n---\n\n".join(render(a) for a in res[:3])
+
     except Exception:
-        pool = [a for a in actions if (not goal or a["goal"]==goal) and (not energy or a["energy"]==energy) and (not max_minutes or a["duration_min"]<=int(max_minutes))]
+        pool = [a for a in actions if (not goal or a["goal"] == goal)
+                and (not energy or a["energy"] == energy)
+                and (not max_minutes or a["duration_min"] <= int(max_minutes))]
         if len(pool) < 3:
             pool = actions[:]
         random.shuffle(pool)
-        out = []
-        for a in pool[:3]:
-            out.append(f"**{a['title']}**  \n{a['instruction']}  \nGoal: {a['goal']} • Duration: {a['duration_min']} min • Energy: {a['energy']} • Context: {', '.join(a['context]()_]()
+        return "\n\n---\n\n".join(render(a) for a in pool[:3])
+
+goals = ["","calm","focus","productivity","wellbeing"]
+energies = ["","low","medium","high"]
+
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")) as demo:
+    gr.Markdown("# ActionMatch\nTop-3 micro-actions based on your situation, goal, energy and time.")
+    with gr.Row():
+        with gr.Column(scale=3):
+            q = gr.Textbox(lines=2, label="Your situation", placeholder="e.g., Stressed before exam")
+            btn = gr.Button("Recommend")
+        with gr.Column(scale=2):
+            g = gr.Dropdown(goals, label="Goal")
+            e = gr.Radio(energies, label="Energy")
+            m = gr.Slider(1, 15, step=1, value=5, label="Max minutes")
+    out = gr.Markdown()
+    btn.click(search, [q, g, e, m], out)
+    gr.Examples(
+        [["Stressed before exam","calm","low",5],
+         ["No energy but need to start studying","focus","low",7],
+         ["Keep switching tabs while writing essay","focus","medium",10]],
+        inputs=[q, g, e, m],
+        outputs=out,
+        fn=search,
+        label="One-click examples",
+        cache_examples=False
+    )
+    demo.queue(concurrency_count=4, max_size=32)
+
+demo.launch()