Update app.py

app.py

@@ -69,32 +69,11 @@ rag_chain = (
 
 import gradio as gr
 
-# # 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.
-    """
-    # Ensure messages is a list of strings
-    if isinstance(messages, list) and all(isinstance(msg, str) for msg in messages):
-        user_message = messages[-1]  # Extract the latest user message
-    else:
-        raise ValueError("Expected messages to be a list of strings.")
-
+# Function to stream responses
+def rag_memory_stream(message, history):
     partial_text = ""
-    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
+    for new_text in rag_chain.stream(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 = [
@@ -133,7 +112,7 @@ Disclaimer: This chatbot is for informational purposes only and not a substitute
 # )
 
 demo = gr.ChatInterface(fn=rag_memory_stream,
-                        type="list",
+                        type="messages",
                         title=title,
                         description=description,
                         fill_height=True,