Update app.py

app.py

@@ -1,7 +1,6 @@
 # ================================
 # 🪞 MoodMirror+ — Text Emotion • Advice-only + brief intros & reasons
-# - Adds an English "Emergency numbers" tab by country
-# - Adds a "Breathing" guided exercise tab
+# - Tabs: Advice • Emergency numbers • Breathing • Journal (with export & delete)
 # ================================
 import os
 import re
@@ -123,10 +122,19 @@ def get_conn():
 
 def init_db():
     conn = get_conn()
+    # sessions
     conn.execute("""CREATE TABLE IF NOT EXISTS sessions(
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         ts TEXT, country TEXT, user_text TEXT, main_emotion TEXT
     )""")
+    # journal
+    conn.execute("""CREATE TABLE IF NOT EXISTS journal(
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        ts TEXT NOT NULL,
+        emotion TEXT,
+        title TEXT,
+        content TEXT
+    )""")
     conn.commit()
     conn.close()
 
@@ -137,6 +145,77 @@ def log_session(country, msg, emotion):
     conn.commit()
     conn.close()
 
+# ---- Journal helpers ----
+def journal_save(title: str, content: str, emotion: str):
+    title = (title or "").strip()
+    content = (content or "").strip()
+    if not content:
+        return False, "Please write something before saving."
+    ts = datetime.utcnow().isoformat(timespec='seconds')
+    conn = get_conn()
+    conn.execute("INSERT INTO journal(ts, emotion, title, content) VALUES (?,?,?,?)",
+        (ts, emotion or "", title, content))
+    conn.commit()
+    conn.close()
+    return True, f"Saved ✓ ({ts} UTC)."
+
+def journal_list(search: str = "", limit: int = 50):
+    q = "SELECT id, ts, emotion, title, content FROM journal"
+    params = []
+    if search:
+        q += " WHERE (LOWER(title) LIKE ? OR LOWER(content) LIKE ? OR LOWER(emotion) LIKE ?)"
+        s = f"%{search.lower()}%"
+        params = [s, s, s]
+    q += " ORDER BY ts DESC LIMIT ?"
+    params.append(int(limit))
+    conn = get_conn()
+    rows = list(conn.execute(q, params))
+    conn.close()
+    options = []
+    table = []
+    for (id_, ts, emo, title, content) in rows:
+        label = f"{ts} — [{(emo or 'neutral')}] {title or (content[:30] + ('…' if len(content) > 30 else ''))}"
+        options.append((label, id_))
+        preview = (content or "").replace("\n", " ")
+        if len(preview) > 120: preview = preview[:120] + "…"
+        table.append([ts, emo or "—", title or "—", preview])
+    return options, table
+
+def journal_get(entry_id: int):
+    conn = get_conn()
+    cur = conn.execute("SELECT ts, emotion, title, content FROM journal WHERE id = ?", (int(entry_id),))
+    row = cur.fetchone()
+    conn.close()
+    if not row: return None
+    ts, emo, title, content = row
+    return {"ts": ts, "emotion": emo or "", "title": title or "", "content": content or ""}
+
+def journal_delete(entry_id: int):
+    conn = get_conn()
+    cur = conn.execute("DELETE FROM journal WHERE id = ?", (int(entry_id),))
+    conn.commit()
+    changes = conn.total_changes
+    conn.close()
+    return changes > 0
+
+def journal_export_txt(entry_id: int):
+    data = journal_get(entry_id)
+    if not data:
+        return None, "Entry not found."
+    fname = f"journal_{entry_id}_{data['ts'].replace(':','-')}.txt"
+    fpath = os.path.join(DATA_DIR, fname)
+    # Build plain text
+    lines = []
+    title = data["title"] or "(Untitled)"
+    lines.append(f"Title: {title}")
+    lines.append(f"Emotion: {data['emotion'] or '-'}")
+    lines.append(f"Saved (UTC): {data['ts']}")
+    lines.append("-" * 40)
+    lines.append(data["content"])
+    with open(fpath, "w", encoding="utf-8") as f:
+        f.write("\n".join(lines))
+    return fpath, f"Ready: {fname}"
+
 # ---------------- Model ----------------
 def load_goemotions_dataset():
     ds = load_dataset("google-research-datasets/go_emotions", "simplified")
@@ -273,8 +352,7 @@ with gr.Blocks(title="🪞 MoodMirror+ — Text Emotion • Advice-only") as dem
 
         # ---- Tab 3: Breathing ----
         with gr.Tab("Breathing"):
-            gr.Markdown("#### 🌬️ Guided breathing\nPick a pattern and cycles, then press **Start**. "
-                        "If you feel dizzy, stop and breathe normally.")
+            gr.Markdown("#### 🌬️ Guided breathing\nPick a pattern and cycles, then press **Start**. If you feel dizzy, stop and breathe normally.")
             with gr.Row():
                 pattern = gr.Dropdown(
                     choices=["4-7-8", "Box (4-4-4-4)", "Coherent (5-5, ~6 breaths/min)"],
@@ -302,6 +380,84 @@ with gr.Blocks(title="🪞 MoodMirror+ — Text Emotion • Advice-only") as dem
                 yield "✅ Done. Notice how your body feels."
             start_btn.click(run_breathing, inputs=[pattern, cycles], outputs=[breathe_out])
 
+        # ---- Tab 4: Journal (with export & delete) ----
+        with gr.Tab("Journal"):
+            gr.Markdown("#### 📝 Personal journal\nWrite freely for a couple of minutes, then save. Entries are stored locally.")
+
+            with gr.Row():
+                j_title = gr.Textbox(label="Title (optional)", placeholder="e.g., A tough day at work")
+                j_emotion = gr.Dropdown(
+                    choices=["neutral","sadness","fear","anger","nervousness","boredom","grief","love","joy","curiosity","gratitude"],
+                    value="neutral", label="Emotion tag"
+                )
+            j_text = gr.Textbox(lines=10, label="Your entry", placeholder="Write whatever you want to get off your chest...")
+
+            with gr.Row():
+                j_save = gr.Button("Save entry", variant="primary")
+                j_status = gr.Markdown()
+
+            gr.Markdown("##### Your entries")
+            with gr.Row():
+                j_search = gr.Textbox(label="Search (title, text or emotion)", placeholder="e.g., anxiety, work, joy")
+                j_refresh = gr.Button("Refresh list")
+            j_entries = gr.Dropdown(label="Entries (newest first)", choices=[], value=None, interactive=True)
+            j_table = gr.Dataframe(headers=["UTC time","Emotion","Title","Preview"], value=[], interactive=False, wrap=True, height=200)
+            j_view = gr.Markdown()
+
+            # Export + Delete UI
+            with gr.Row():
+                j_export_btn = gr.Button("Export .txt")
+                j_export_file = gr.File(label="Download exported entry", visible=True)
+                j_delete_btn = gr.Button("Delete entry", variant="stop")
+
+            # Backend glue
+            def _refresh_entries(search):
+                options, table = journal_list(search or "", 50)
+                return gr.Dropdown(choices=options, value=None), table
+
+            def _save_entry(title, text, emotion, search):
+                ok, msg = journal_save(title, text, emotion)
+                drop, table = _refresh_entries(search)
+                clear_text = "" if ok else text
+                clear_title = "" if ok else title
+                return msg, drop, table, clear_text, clear_title
+
+            def _load_entry(entry_id):
+                if entry_id is None: return ""
+                data = journal_get(entry_id)
+                if not data: return "Entry not found."
+                title_line = f"### {data['title']}" if data['title'] else "### (Untitled)"
+                emo_line = f"**Emotion:** {data['emotion'] or '—'}  \n**Saved (UTC):** {data['ts']}"
+                return f"{title_line}\n\n{emo_line}\n\n---\n\n{data['content']}"
+
+            def _export_entry(entry_id):
+                if entry_id is None:
+                    return None, "Select an entry to export."
+                path, msg = journal_export_txt(int(entry_id))
+                return path, msg
+
+            def _delete_entry(entry_id, search):
+                if entry_id is None:
+                    status = "Select an entry to delete."
+                    drop, table = _refresh_entries(search)
+                    return status, drop, table, ""
+                ok = journal_delete(int(entry_id))
+                status = "Deleted ✓" if ok else "Entry not found."
+                drop, table = _refresh_entries(search)
+                return status, drop, table, ""
+
+            # Wire actions
+            j_save.click(_save_entry, inputs=[j_title, j_text, j_emotion, j_search],
+                         outputs=[j_status, j_entries, j_table, j_text, j_title])
+            j_refresh.click(_refresh_entries, inputs=[j_search], outputs=[j_entries, j_table])
+            j_search.submit(_refresh_entries, inputs=[j_search], outputs=[j_entries, j_table])
+            j_entries.change(_load_entry, inputs=[j_entries], outputs=[j_view])
+
+            j_export_btn.click(_export_entry, inputs=[j_entries], outputs=[j_export_file, j_status])
+            j_delete_btn.click(_delete_entry, inputs=[j_entries, j_search],
+                               outputs=[j_status, j_entries, j_table, j_view])
+
 if __name__ == "__main__":
+    gr.set_theme("default")
     demo.queue()
     demo.launch()