Update app.py

app.py

@@ -21,36 +21,28 @@ profile = load_json(PROFILE_FILE, {})
 daily = load_json(DAILY_FILE, {})
 
 def build_context(profile, daily):
-    recent_days = sorted(daily.keys(), reverse=True)[:7]
-    daily_lines = "\n".join([f"{d}: {daily[d].get('log','')}" for d in recent_days])
-    context = (
-        f"Profile:\n{json.dumps(profile, indent=2)}\n"
-        f"Recent daily logs:\n{daily_lines}\n"
-        "You are a helpful assistant. Answer only using the provided information. "
-        "If you don't know the answer, reply: 'Sheetal hasn't shared that yet!'"
-    )
-    return context
-
-def get_llm():
-    return pipeline(
-        "text-generation",
-        model="gpt2",
-        max_new_tokens=128,
-        do_sample=True,
-        temperature=0.7,
-    )
+    # Combine profile info and last 7 days logs
+    context = []
+    for k, v in profile.items():
+        context.append(f"{k.capitalize()}: {v}")
+    for d in sorted(daily.keys(), reverse=True)[:7]:
+        context.append(f"On {d}: {daily[d].get('log','')}")
+    return "\n".join(context)
 
-llm = None
+# QA Pipeline - loads once and is reused
+qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
 def chatbot_qa(user_q):
-    global llm
-    if llm is None:
-        llm = get_llm()
     context = build_context(profile, daily)
-    prompt = f"System: {context}\nUser: {user_q}\nAssistant:"
-    outputs = llm(prompt, max_new_tokens=128)
-    answer = outputs[0]["generated_text"].split("Assistant:")[-1].strip()
-    return answer
+    try:
+        result = qa_pipeline(question=user_q, context=context)
+        answer = result["answer"].strip()
+        # If the model cannot answer, it often returns '' or 'empty'
+        if not answer or answer.lower() in ["empty", ""]:
+            return "Sheetal hasn't shared that yet!"
+        return answer
+    except Exception as e:
+        return f"Error: {e}"
 
 def save_profile_and_log(name, city, job, about, freeform):
     today = str(date.today())
@@ -85,14 +77,17 @@ def check_password(password):
     else:
         return gr.update(visible=False), gr.update(value="❌ Access denied. Try again.", visible=True)
 
+def show_profile():
+    return json.dumps(profile, indent=2), json.dumps(daily, indent=2)
+
 with gr.Blocks(title="Sheetal's Chatbot") as demo:
     gr.Markdown("# 🌸 Sheetal's Personal Chatbot")
     gr.Markdown("Ask anything about Sheetal!")
 
     with gr.Tab("📝 Admin (Sheetal)"):
         admin_password = gr.Textbox(label="Enter admin password", type="password", max_lines=1)
-        password_status = gr.Textbox(label="Password status", value="", interactive=False, visible=False)
         unlock_panel = gr.Button("Unlock Admin Panel")
+        password_status = gr.Textbox(label="Password status", value="", interactive=False, visible=False)
 
         with gr.Column(visible=False) as admin_panel:
             with gr.Row():
@@ -109,6 +104,11 @@ with gr.Blocks(title="Sheetal's Chatbot") as demo:
                 inputs=[name, city, job, about, today_log],
                 outputs=[admin_status, logs_output]
             )
+            # Debug: show current data
+            show_btn = gr.Button("Show Current Data")
+            profile_out = gr.Textbox(label="Profile JSON")
+            daily_out = gr.Textbox(label="Daily JSON")
+            show_btn.click(fn=show_profile, outputs=[profile_out, daily_out])
 
         unlock_panel.click(
             fn=check_password,