Update app.py

app.py

@@ -30,7 +30,7 @@ def build_context(profile, daily):
 def get_llm():
     return pipeline(
         "text-generation",
-        model="google/flan-t5-small",  # Fast!
+        model="google/flan-t5-small",
         max_new_tokens=128,
         do_sample=True,
         temperature=0.7,
@@ -48,19 +48,73 @@ def chatbot_qa(user_q):
     answer = outputs[0]["generated_text"].split("Assistant:")[-1].strip()
     return answer
 
-with gr.Blocks(title="Sheetal's Personal Chatbot") as demo:
-    gr.Markdown("# 🌸 Sheetal's Personal Chatbot")
-    gr.Markdown("Ask anything about Sheetal!")
-
-    with gr.Tab("💬 Ask About Sheetal"):
-        gr.Markdown("### 💬 Ask Anything About Sheetal")
-        user_q = gr.Textbox(label="Type your question here:")
-        ask_btn = gr.Button("Ask")
-        answer_box = gr.Textbox(label="Bot answer", interactive=False, lines=2, max_lines=4)
-        ask_btn.click(
-            fn=chatbot_qa,
-            inputs=user_q,
-            outputs=answer_box
-        )
-
-demo.launch()
+def save_profile_and_log(name, city, job, about, freeform):
+    today = str(date.today())
+    profile.update({
+        "name": name,
+        "city": city,
+        "job": job,
+        "about": about
+    })
+    save_json(PROFILE_FILE, profile)
+    daily[today] = {"log": freeform}
+    save_json(DAILY_FILE, daily)
+    logs = ""
+    for d in sorted(daily.keys(), reverse=True)[:5]:
+        logs += f"**{d}**: {daily[d]['log']}\n"
+    return (
+        "✅ Profile & log updated!",
+        logs
+    )
+
+def recent_logs():
+    logs = ""
+    for d in sorted(daily.keys(), reverse=True)[:5]:
+        logs += f"**{d}**: {daily[d]['log']}\n"
+    return logs
+
+SECRET_CODE = "1234"  # Change this to your secret
+
+def build_ui(request: gr.Request):
+    # Check for secret in URL, e.g., ?admin=1234
+    query_params = dict(request.query_params)
+    show_admin = query_params.get("admin", [None])[0] == SECRET_CODE
+
+    with gr.Blocks(title="Sheetal's Chatbot") as demo:
+        gr.Markdown("# 🌸 Sheetal's Personal Chatbot")
+        gr.Markdown("Ask anything about Sheetal!")
+
+        if show_admin:
+            with gr.Tab("📝 Admin (Sheetal)"):
+                with gr.Row():
+                    name = gr.Textbox(label="Name", value=profile.get("name", "Sheetal"), max_lines=1)
+                    city = gr.Textbox(label="City", value=profile.get("city", ""), max_lines=1)
+                    job = gr.Textbox(label="Profession", value=profile.get("job", ""), max_lines=1)
+                about = gr.Textbox(label="About You", value=profile.get("about", ""), lines=2, max_lines=3)
+                today_log = gr.Textbox(label="What do you want to remember about today?", value=daily.get(str(date.today()), {}).get("log", ""), lines=2, max_lines=3)
+                save_btn = gr.Button("💾 Save Profile & Today's Log")
+                admin_status = gr.Markdown("")
+                logs_output = gr.Markdown(recent_logs())
+                save_btn.click(
+                    fn=save_profile_and_log,
+                    inputs=[name, city, job, about, today_log],
+                    outputs=[admin_status, logs_output]
+                )
+
+        with gr.Tab("💬 Ask About Sheetal"):
+            user_q = gr.Textbox(label="Type your question here:")
+            ask_btn = gr.Button("Ask")
+            answer_box = gr.Textbox(label="Bot answer", interactive=False, lines=2, max_lines=4)
+            ask_btn.click(
+                fn=chatbot_qa,
+                inputs=user_q,
+                outputs=answer_box
+            )
+    return demo
+
+gr.mount_gradio_app(app=build_ui, path="/", root_path="")
+
+# For Hugging Face Spaces, use this launch line:
+demo = build_ui
+if __name__ == "__main__":
+    demo.launch()