Update app.py

app.py

@@ -298,35 +298,30 @@ def chatbot_interface(message, history, model, temperature, num_calls, use_web_s
         else:
             return history  # Return if both text and audio are empty
         
-        history.append((input_text, ""))
+        # Add user message to history
+        history.append((input_text, None))
         
-        for response in respond(input_text, history, model, temperature, num_calls, use_web_search, selected_docs):
-            # Ensure the response is a string
-            if isinstance(response, (list, tuple)):
-                response = ' '.join(str(item) for item in response if item is not None)
-            elif not isinstance(response, str):
-                response = str(response)
-            
-            # Truncate response if it's too long
-            max_length = 32000  # Adjust this value as needed
-            if len(response) > max_length:
-                response = response[:max_length] + "... (truncated)"
-            
+        # Get response from the model
+        response_generator = respond(input_text, history, model, temperature, num_calls, use_web_search, selected_docs)
+        
+        full_response = ""
+        for partial_response in response_generator:
+            full_response += str(partial_response)
             # Update the last message in history
-            if history:
-                history[-1] = (input_text, response)
-            else:
-                history.append((input_text, response))
-            
-            # Yield a copy of the history to avoid reference issues
-            yield list(history)
+            history[-1] = (input_text, full_response)
+            yield history
+        
+        # If no response was generated, yield an error message
+        if not full_response:
+            history[-1] = (input_text, "I'm sorry, I couldn't generate a response. Please try again.")
+            yield history
+
     except Exception as e:
         logging.error(f"Error in chatbot_interface: {str(e)}")
         error_message = f"An error occurred: {str(e)}"
-        if not isinstance(history, list):
-            history = []
         history.append((input_text if 'input_text' in locals() else "Error", error_message))
-        yield list(history)
+        yield history
+
 def retry_last_response(history, model, temperature, num_calls):
     if not history:
         return history