Update app.py

app.py

@@ -33,39 +33,41 @@ def respond(
             temperature=temperature,
             top_p=top_p,
         ):
-            if isinstance(message, ChatCompletionStreamOutput):
-                # Extract the content from the ChatCompletionStreamOutput object
-                content = message.choices[0].delta.content
-                if content is not None:
-                    response += content
-                    yield response
-                # Check if this is the last message in the stream
-                if message.choices[0].finish_reason == 'stop':
-                    break
-            elif isinstance(message, dict):
-                # Handle dictionary responses (possible API format change)
-                content = message.get('choices', [{}])[0].get('delta', {}).get('content')
-                if content:
-                    response += content
-                    yield response
-                if message.get('choices', [{}])[0].get('finish_reason') == 'stop':
-                    break
-            elif isinstance(message, str):
-                # Handle string responses (possible raw content)
-                response += message
-                yield response
-            else:
-                print(f"Unexpected message type: {type(message)}")
-                print(f"Message content: {message}")
+            try:
+                if isinstance(message, ChatCompletionStreamOutput):
+                    content = message.choices[0].delta.content
+                    if content is not None:
+                        response += content
+                        yield response
+                    if message.choices[0].finish_reason == 'stop':
+                        break
+                elif isinstance(message, dict):
+                    content = message.get('choices', [{}])[0].get('delta', {}).get('content')
+                    if content:
+                        response += content
+                        yield response
+                    if message.get('choices', [{}])[0].get('finish_reason') == 'stop':
+                        break
+                elif isinstance(message, str):
+                    if message.strip():  # Only process non-empty strings
+                        response += message
+                        yield response
+                else:
+                    print(f"Unexpected message type: {type(message)}")
+                    print(f"Message content: {message}")
+            except Exception as e:
+                print(f"Error processing message: {e}")
+                print(f"Problematic message: {message}")
+                continue  # Continue to the next message even if there's an error
         
         # Final yield to ensure all content is returned
         if response:
             yield response
 
     except Exception as e:
-        print(f"An error occurred: {e}")
+        print(f"An error occurred in the main loop: {e}")
         if response:
-            yield response + f"\n\nAn error occurred: {e}"
+            yield response
         else:
             yield f"An error occurred: {e}"