Update app.py

app.py

@@ -25,8 +25,16 @@ def proxy(endpoint):
     data = request.get_json()
     response = requests.post(f"{BASE_URL}/{endpoint}", headers=headers, json=data)
 
-    # Return the OpenAI response
-    return jsonify(response.json()), response.status_code
+    # Handle non-JSON responses
+    try:
+        response_json = response.json()
+    except requests.exceptions.JSONDecodeError:
+        # Log the raw response text for debugging
+        print("Non-JSON response:", response.text)
+        return jsonify({"error": "Invalid response from OpenAI API", "details": response.text}), response.status_code
+
+    # Return the OpenAI response if it's JSON
+    return jsonify(response_json), response.status_code
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000)