UPDATE APP.PY ADDED CHAT_HISTORY FEATURE

app.py

@@ -1,32 +1,33 @@
-# app.py
-import os
 import gradio as gr
 from pipeline import run_with_chain
+from my_memory_logic import memory, restatement_chain
 
-def ask_dailywellness(query: str) -> str:
-    """
-    Calls our main pipeline function from pipeline.py
-    """
-    return run_with_chain(query)
+def chat_history_fn(user_input, history):
+    # Convert `history` (list of [user, ai] pairs) into memory messages
+    for user_msg, ai_msg in history:
+        memory.chat_memory.add_user_message(user_msg)
+        memory.chat_memory.add_ai_message(ai_msg)
+    
+    # 1) Restate the user question with chat history
+    reformulated_q = restatement_chain.run({
+        "chat_history": memory.chat_memory.messages,
+        "input": user_input
+    })
+    
+    # 2) Pass the reformulated question to your pipeline
+    answer = run_with_chain(reformulated_q)
+    
+    # 3) Update memory
+    memory.chat_memory.add_user_message(user_input)
+    memory.chat_memory.add_ai_message(answer)
+    
+    # 4) Return the new message + updated history for Gradio
+    history.append((user_input, answer))
+    return history, history
 
-# Build a simple Gradio interface
-interface = gr.Interface(
-    fn=ask_dailywellness,
-    inputs=gr.Textbox(
-        lines=2,
-        label="Ask DailyWellnessAI",
-        placeholder="e.g., How do I improve my posture while working?"
-    ),
-    outputs=gr.Textbox(
-        label="DailyWellnessAI Answer"
-    ),
-    title="DailyWellnessAI Chat",
-    description=(
-        "Ask a wellness question, or something about DailyWellnessAI (brand). "
-        "If the question is out of scope, you'll get a polite refusal."
-    ),
-    allow_flagging="never"
+demo = gr.ChatInterface(
+    fn=chat_history_fn,
+    title="DailyWellnessAI with Memory",
+    description="A chat bot that remembers context using memory + question restatement."
 )
-
-if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch()