Update app.py

app.py

@@ -221,8 +221,12 @@ def respond(message, history, model, temperature, num_calls, use_web_search):
 
     try:
         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}" if sources else main_content
+            for main_content, sources, search_results in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
+                if sources:
+                    formatted_sources = match_and_format_sources(sources, search_results)
+                    response = f"{main_content}\n\n{formatted_sources}"
+                else:
+                    response = main_content
                 first_line = response.split('\n')[0] if response else ''
                 logging.info(f"Generated Response (first line): {first_line}")
                 yield response
@@ -238,10 +242,11 @@ def respond(message, history, model, temperature, num_calls, use_web_search):
                 else:
                     context_str = "No documents available."
                 
-                for partial_response in get_response_from_cloudflare(prompt="", context=context_str, query=message, num_calls=num_calls, temperature=temperature, search_type="pdf"):
-                    first_line = partial_response.split('\n')[0] if partial_response else ''
+                for main_content, sources in get_response_from_cloudflare(prompt="", context=context_str, query=message, num_calls=num_calls, temperature=temperature, search_type="pdf"):
+                    response = f"{main_content}\n\n{sources}" if sources else main_content
+                    first_line = response.split('\n')[0] if response else ''
                     logging.info(f"Generated Response (first line): {first_line}")
-                    yield partial_response
+                    yield response
             else:
                 # Use Hugging Face API
                 for partial_response in get_response_from_pdf(message, model, num_calls=num_calls, temperature=temperature):