Update app.py

app.py

@@ -69,16 +69,32 @@ rag_chain = (
 
 import gradio as gr
 
-# Function to stream responses
-def rag_memory_stream(message, history):
+# # Function to stream responses
+# def rag_memory_stream(message, history):
+#     partial_text = ""
+#     for new_text in rag_chain.stream(message):  # Assuming rag_chain is pre-defined
+#         partial_text += new_text
+#         yield partial_text
+
+# Example rag_memory_stream function with history handling
+def rag_memory_stream(messages, history=[]):
+    """
+    A generator-based function that processes messages, maintains history,
+    and streams responses for interaction with the chatbot.
+    """
     partial_text = ""
-    for new_text in rag_chain.stream(message):  # Assuming rag_chain is pre-defined
+    user_message = messages[-1]["content"]  # Extract the latest user message
+    history.append({"user": user_message, "bot": ""})  # Add to history
+
+    # Simulate response generation (replace with actual rag_chain logic)
+    for new_text in rag_chain.stream(user_message):  # Assuming rag_chain is pre-defined
         partial_text += new_text
+        history[-1]["bot"] = partial_text  # Update bot response in history
         yield partial_text
 
 examples = [
-    "What are the side effects of aspirin?",
-    "What is the disease for constant fatigue and muscle weakness?"
+    [{"role": "user", "content": "What are the side effects of aspirin?"}],
+    [{"role": "user", "content": "Can ibuprofen cause dizziness?"}],
 ]
 
 # Title and description for the app