Update app.py

app.py

@@ -38,9 +38,16 @@ def respond(
         ):
             token = message.choices[0].delta.content
 
+            # Check if the token is a valid JSON string
+            try:
+                json.loads(token)
+            except json.JSONDecodeError:
+                yield f"Error: Invalid JSON response - {token}"
+                break
+
             response += token
             yield response
-    except json.JSONDecodeError as e:
+    except Exception as e:
         yield f"Error: {e}"
 
 """