Update app.py

app.py

@@ -271,8 +271,7 @@ with gr.Blocks() as demo:
                 formatted_response = f"{main_content}\n\nSources:\n{sources}"
             else:
                 print("Using PDF search")
-                response = get_response_from_pdf(message, model, num_calls=num_calls, temperature=temperature, stop_clicked=stop_clicked)
-                formatted_response = response
+                formatted_response = get_response_from_pdf(message, model, num_calls=num_calls, temperature=temperature, stop_clicked=stop_clicked)
     
             print(f"Generated response: {formatted_response[:100]}...")
     
@@ -282,14 +281,13 @@ with gr.Blocks() as demo:
         
         is_generating = False
         print(f"Returning final response")
-        return formatted_response, is_generating, stop_clicked
+        return "", history + [(message, 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}")
-        response, new_is_generating, new_stop_clicked = protected_generate_response(
+        _, new_history, 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