Update app.py

app.py

@@ -100,6 +100,8 @@ def create_web_search_vectors(search_results):
     
     return FAISS.from_documents(documents, embed)
 
+import json
+
 def get_response_with_search(query, model, num_calls=3, temperature=0.2):
     search_results = duckduckgo_search(query)
     web_search_database = create_web_search_vectors(search_results)
@@ -132,16 +134,38 @@ After writing the document, please provide a list of sources used in your respon
     
     main_content = ""
     for i in range(num_calls):
-        for message in client.chat_completion(
-            messages=[{"role": "user", "content": prompt}],
-            max_tokens=max_new_tokens,
-            temperature=temperature,
-            stream=False,
-        ):
-            if message.choices 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
+        try:
+            response = client.chat_completion(
+                messages=[{"role": "user", "content": prompt}],
+                max_new_tokens=max_new_tokens,
+                temperature=temperature,
+                stream=False,
+            )
+            
+            # Log the raw response for debugging
+            logging.info(f"Raw API response: {response}")
+            
+            # Check if the response is a string (which might be an error message)
+            if isinstance(response, str):
+                logging.error(f"API returned an unexpected string response: {response}")
+                yield f"An error occurred: {response}", ""
+                return
+            
+            # If it's not a string, assume it's the expected object structure
+            if hasattr(response, 'choices') and response.choices:
+                for choice in response.choices:
+                    if hasattr(choice, 'message') and hasattr(choice.message, 'content'):
+                        chunk = choice.message.content
+                        main_content += chunk
+                        yield main_content, ""  # Yield partial main content without sources
+            else:
+                logging.error(f"Unexpected response structure: {response}")
+                yield "An unexpected error occurred. Please try again.", ""
+                
+        except Exception as e:
+            logging.error(f"Error in API call: {str(e)}")
+            yield f"An error occurred: {str(e)}", ""
+            return
 
 def vote(data: gr.LikeData):
     if data.liked: