Update app.py

app.py

@@ -292,7 +292,7 @@ def chatbot_interface(message, history, use_web_search, model, temperature, num_
             history[-1] = (message, "Generating response... (This may take a moment)")
             yield history
         
-        for response in respond(message, history, model, temperature, num_calls, use_web_search):
+        for response in respond(message, history, model, temperature, num_calls, use_web_search, selected_docs, instruction_key):
             history[-1] = (message, response)
             yield history
     except gr.CancelledError:
@@ -327,8 +327,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
             use_web_search = False  # Ensure we use PDF search for summaries
 
         if use_web_search:
-            for main_content, sources in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
-                response = f"{main_content}\n\n{sources}"
+            for response, _ in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
                 yield response
         else:
             embed = get_embeddings()
@@ -520,13 +519,16 @@ Critique:
 Revised Response:"""
 
         # Generate final response
+        full_response = ""
         for chunk in client.text_generation(final_prompt, max_new_tokens=1500, temperature=temperature, stream=True):
-            yield chunk, ""
+            full_response += chunk
+            yield full_response, ""
 
         # Add a disclaimer
         disclaimer = ("\nNote: This response was generated by an AI model based on web search results. "
                       "While efforts have been made to ensure accuracy, please verify important information from authoritative sources.")
-        yield disclaimer, ""
+        full_response += disclaimer
+        yield full_response, ""
     
     except Exception as e:
         logging.error(f"Error in multi-step generation process: {str(e)}")