Update app.py

app.py

@@ -484,16 +484,20 @@ Write a detailed and complete response that answers the following user question:
         response = ""
         for i in range(num_calls):
             logging.info(f"API call {i+1}/{num_calls}")
-            for message in client.chat_completion(
-                messages=[{"role": "user", "content": prompt}],
-                max_tokens=10000,
-                temperature=temperature,
-                stream=True,
-            ):
-                if message.choices and message.choices[0].delta and message.choices[0].delta.content:
-                    chunk = message.choices[0].delta.content
-                    response += chunk
-                    yield response  # Yield partial response
+            try:
+                for message in client.chat_completion(
+                    messages=[{"role": "user", "content": prompt}],
+                    max_tokens=10000,
+                    temperature=temperature,
+                    stream=True,
+                ):
+                    if message.choices and message.choices[0].delta and message.choices[0].delta.content:
+                        chunk = message.choices[0].delta.content
+                        response += chunk
+                        yield response  # Yield partial response
+                logging.info(f"API call {i+1} completed successfully")
+            except Exception as e:
+                logging.error(f"Error in API call {i+1}: {str(e)}")
         
         logging.info("Finished generating response")