Update app.py

app.py

@@ -478,9 +478,9 @@ def get_response_with_search(query, model, num_calls=3, temperature=0.2):
     context = retrieve_web_search_results(database, query)
     
     prompt = f"""Using the following context from web search results:
-{context}
-Write a detailed and complete research document that fulfills the following user request: '{query}'
-After writing the document, please provide a list of sources used in your response."""
+    {context}
+    Write a detailed and complete research document that fulfills the following user request: '{query}'
+    After writing the document, please provide a list of sources used in your response."""
 
     try:
         if model == "@cf/meta/llama-3.1-8b-instruct":
@@ -499,10 +499,13 @@ After writing the document, please provide a list of sources used in your respon
                     temperature=temperature,
                     stream=True,
                 ):
-                    if message.choices and message.choices[0].delta and message.choices[0].delta.content:
+                    if message.choices and len(message.choices) > 0 and message.choices[0].delta and message.choices[0].delta.content:
                         chunk = message.choices[0].delta.content
                         main_content += chunk
                         yield main_content, ""  # Yield partial main content without sources
+    except Exception as e:
+        logging.error(f"Error in get_response_with_search: {str(e)}", exc_info=True)
+        yield f"An error occurred while processing the search results: {str(e)}", ""
     finally:
         # Clean up the temporary database
         cleanup_web_search_database(temp_dir)