Update app.py

app.py

@@ -69,8 +69,8 @@ def stream_chat(
 ):
     # Prepare the conversation context
     conversation_text = system_prompt + "\n"
-    for prompt, answer in history:
-        conversation_text += f"User: {prompt}\nMisMath: {answer}\n"
+    for _, answer in history:
+        conversation_text += f"MisMath: {answer}\n"
 
     conversation_text += f"User: {message}\nMisMath:"
 
@@ -95,14 +95,13 @@ def stream_chat(
         thread.start()
 
     buffer = ""
+    final_output = ""
     for new_text in streamer:
         buffer += new_text
-        # Clean the buffer to remove unwanted prefixes
+        # Extract only the final response after the last "MisMath:"
         if "MisMath:" in buffer:
-            cleaned_text = buffer.split("MisMath:")[-1].strip()
-            yield cleaned_text
-        else:
-            yield buffer.strip()
+            final_output = buffer.split("MisMath:")[-1].strip()
+        yield final_output
 
 # Define the Gradio chatbot component
 chatbot = gr.Chatbot(height=500, placeholder=PLACEHOLDER)