Update app.py

app.py

@@ -5,6 +5,7 @@ from io import BytesIO
 from PIL import Image
 import numpy as np
 import traceback
+import json
 
 def generate_gradio_app(api_key, image):
     if not api_key:
@@ -74,8 +75,22 @@ Please generate the Gradio code based on the provided image, focusing on the mos
     except Exception as e:
         error_message = str(e)
         stack_trace = traceback.format_exc()
+        
+        # Check if the error is related to the API response
+        if "TogetherErrorResponse" in error_message:
+            try:
+                # Try to parse the error message as JSON
+                error_data = json.loads(error_message.split("TogetherErrorResponse")[-1].strip())
+                if isinstance(error_data.get('code'), int):
+                    error_data['code'] = str(error_data['code'])  # Convert code to string
+                error_message = f"API Error: {error_data.get('message', 'Unknown error')}"
+            except json.JSONDecodeError:
+                # If parsing fails, use the original error message
+                pass
+        
         return f"An error occurred: {error_message}\n\nStack trace:\n{stack_trace}\n\nPlease check your API key and try again."
 
+# The rest of your code remains the same
 with gr.Blocks() as demo:
     gr.Markdown("# Generate Concise Gradio App from Wireframe")
     gr.Markdown("Enter your Together API key, upload an image of your UI design, and we'll generate a compact Gradio code to recreate its main elements.")