Update app.py

app.py

@@ -258,7 +258,6 @@ with gr.Blocks() as demo:
     
         is_generating = True
         
-        # Handle both boolean and Gradio State object
         if isinstance(stop_clicked, gr.State):
             stop_clicked.value = False
         else:
@@ -277,24 +276,20 @@ with gr.Blocks() as demo:
     
             print(f"Generated response: {formatted_response[:100]}...")
     
-            if not (isinstance(stop_clicked, gr.State) and stop_clicked.value) and not stop_clicked:
-                print("Appending to history")
-                history.append((message, formatted_response))
-            else:
-                print("Stop clicked, not appending to history")
         except Exception as e:
             print(f"Error generating response: {str(e)}")
-            history.append((message, "I'm sorry, but I encountered an error while generating the response. Please try again."))
+            formatted_response = "I'm sorry, but I encountered an error while generating the response. Please try again."
         
         is_generating = False
-        print(f"Returning history with {len(history)} items")
-        return "", history, is_generating, stop_clicked
-    
+        print(f"Returning final response")
+        return formatted_response, is_generating, stop_clicked
+
     def on_submit(message, history, use_web_search, model, temperature, num_calls, is_generating, stop_clicked):
         print(f"Submit button clicked with message: {message}")
-        _, new_history, new_is_generating, new_stop_clicked = protected_generate_response(
+        response, new_is_generating, new_stop_clicked = protected_generate_response(
             message, history, use_web_search, model, temperature, num_calls, is_generating, stop_clicked
         )
+        new_history = history + [(message, response)]
         print(f"New history has {len(new_history)} items")
         return "", new_history, new_is_generating, new_stop_clicked