Update app.py

app.py

@@ -5,6 +5,7 @@ import base64
 from io import BytesIO
 from PIL import Image
 import numpy as np
+import json
 
 # Initialize the Together client
 api_key = os.environ.get('TOGETHER_API_KEY')
@@ -31,24 +32,13 @@ def generate_gradio_app(image):
         img_str = base64.b64encode(buffered.getvalue()).decode()
         
         # Prepare the messages for the API call
-        system_message = """You are an AI assistant that can analyze wireframe images and generate detailed Gradio code based on their content. Your task is to provide a complete, runnable Gradio application that recreates the UI elements seen in the wireframe."""
-
+        system_message = "You are an AI assistant that can analyze wireframe images and generate detailed Gradio code based on their content."
         user_message = f"""
         <image>
         data:image/png;base64,{img_str}
         </image>
         
-        Analyze this wireframe image and generate a complete Python code using Gradio that recreates all the main elements seen in the image. Follow these guidelines:
-        1. Use appropriate Gradio components that best represent each UI element in the wireframe.
-        2. Include all necessary imports at the beginning of the code.
-        3. Implement placeholder functions for any interactive elements (buttons, inputs, etc.).
-        4. Use gr.Blocks() to create a layout that matches the wireframe as closely as possible.
-        5. Add appropriate labels and descriptions for all components.
-        6. Include the gr.Blocks().launch() call at the end of the code.
-        7. Provide a complete, runnable Gradio application that can be executed as-is.
-        8. Add comments explaining the purpose of each major section or component.
-
-        Please generate the entire code, including all necessary parts to make it a fully functional Gradio application.
+        Analyze this wireframe image and generate a complete Python code using Gradio that recreates all the main elements seen in the image. Provide a complete, runnable Gradio application.
         """
         
         messages = [
@@ -60,24 +50,29 @@ def generate_gradio_app(image):
         response = client.chat.completions.create(
             model="meta-llama/Llama-Vision-Free",
             messages=messages,
-            max_tokens=2048,
+            max_tokens=5000,
             temperature=0.7,
             top_p=0.7,
             top_k=50,
             repetition_penalty=1,
-            stop=["<|im_end|>"],
-            stream=True
         )
         
-        # Collect the streamed response
-        generated_code = ""
-        for chunk in response:
-            if chunk.choices[0].delta.content is not None:
-                generated_code += chunk.choices[0].delta.content
-        
+        # Extract the generated code from the response
+        generated_code = response.choices[0].message.content
         return generated_code
+
     except Exception as e:
-        return f"An error occurred: {str(e)}\n\nPlease try again or check your API key and connection."
+        error_message = str(e)
+        if "400" in error_message:
+            return "Error 400: Bad Request. The API request was invalid. Please check your input and try again."
+        elif "401" in error_message:
+            return "Error 401: Unauthorized. Please check your API key and ensure it's correctly set."
+        elif "429" in error_message:
+            return "Error 429: Too Many Requests. Please wait a moment and try again."
+        elif "500" in error_message:
+            return "Error 500: Internal Server Error. There's an issue with the API service. Please try again later."
+        else:
+            return f"An unexpected error occurred: {error_message}\nPlease try again or contact support if the issue persists."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Turn your wireframe into a Gradio app")