Update app.py

app.py

@@ -435,13 +435,12 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
     logging.info(f"Model Used: {model}")
     logging.info(f"Selected Documents: {selected_docs}")
     logging.info(f"Use Web Search: {use_web_search}")
-
+    
     if use_web_search:
         original_query = message
         rephrased_query = rephrase_query(message, conversation_manager)
         logging.info(f"Original query: {original_query}")
         logging.info(f"Rephrased query: {rephrased_query}")
-
         final_summary = ""
         for _ in range(num_calls):
             search_results = get_web_search_results(rephrased_query)
@@ -451,8 +450,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
                 final_summary += search_results[0]["error"] + "\n\n"
             else:
                 summary = summarize_web_results(rephrased_query, search_results, conversation_manager)
-                final_summary += summary + "\n\n"
-
+                final_summary += str(summary) + "\n\n"
         if final_summary:
             conversation_manager.add_interaction(original_query, final_summary)
             yield final_summary
@@ -483,21 +481,21 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
             if model.startswith("duckduckgo/"):
                 # Use DuckDuckGo chat with context
                 for partial_response in get_response_from_duckduckgo(message, model, context_str, num_calls, temperature):
-                    yield partial_response
+                    yield str(partial_response)
             elif model == "@cf/meta/llama-3.1-8b-instruct":
                 # Use Cloudflare API
                 for partial_response in get_response_from_cloudflare(prompt="", context=context_str, query=message, num_calls=num_calls, temperature=temperature, search_type="pdf"):
-                    yield partial_response
+                    yield str(partial_response)
             else:
                 # Use Hugging Face API
                 for partial_response in get_response_from_pdf(message, model, selected_docs, num_calls=num_calls, temperature=temperature):
-                    yield partial_response
+                    yield str(partial_response)
         except Exception as e:
             logging.error(f"Error with {model}: {str(e)}")
             if "microsoft/Phi-3-mini-4k-instruct" in model:
                 logging.info("Falling back to Mistral model due to Phi-3 error")
                 fallback_model = "mistralai/Mistral-7B-Instruct-v0.3"
-                yield from respond(message, history, fallback_model, temperature, num_calls, selected_docs)
+                yield from (str(response) for response in respond(message, history, fallback_model, temperature, num_calls, selected_docs))
             else:
                 yield f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model."