Fix chatbot response format for streaming output

app.py

@@ -75,11 +75,11 @@ async def research_assistant(query):
 
         for chunk in stream:
             full_response += chunk
-            yield format_output(full_response)
+            yield full_response  # Yield plain string for chatbot processing
 
         citations = generate_citations(search_results)
         citation_text = format_citations(citations)
-        full_output = format_output(full_response) + citation_text
+        full_output = full_response + citation_text
 
         cache_result(query, full_output)
         server_monitor.report_success()
@@ -161,7 +161,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Research Assistant") as demo:
             """)
             
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(height=500, label="Research Conversation")
+            chatbot = gr.Chatbot(height=500, label="Research Conversation", latex_delimiters=[{"left": "$$", "right": "$$", "display": True}])
             msg = gr.Textbox(
                 label="Research Question",
                 placeholder="Ask a complex research question...",
@@ -183,10 +183,17 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI Research Assistant") as demo:
             )
     
     def respond(message, chat_history):
-        bot_response = ""
+        # Add user message to chat history
+        chat_history.append((message, ""))
+        yield chat_history
+        
+        # Get streaming response
+        full_response = ""
         for partial_response in research_assistant_wrapper(message):
-            bot_response = partial_response
-            yield bot_response
+            full_response = partial_response
+            # Update the last message in chat history with the partial response
+            chat_history[-1] = (message, full_response)
+            yield chat_history
     
     submit_btn.click(respond, [msg, chatbot], chatbot)
     msg.submit(respond, [msg, chatbot], chatbot)