Update app.py

app.py

@@ -31,12 +31,18 @@ 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 Gradio code based on their content."
+        user_message = f"""
+        <image>
+        data:image/png;base64,{img_str}
+        </image>
+        
+        Analyze this wireframe image and generate Python code using Gradio that could recreate the main elements seen in the image. Use Gradio components that best represent the UI elements in the wireframe.
+        """
+        
         messages = [
-            {"role": "system", "content": "You are an AI assistant that can analyze wireframe images and generate Gradio code based on their content."},
-            {"role": "user", "content": [
-                {"type": "image_url", "image_url": f"data:image/png;base64,{img_str}"},
-                {"type": "text", "text": "Analyze this wireframe image and generate Python code using Gradio that could recreate the main elements seen in the image. Use Gradio components that best represent the UI elements in the wireframe."}
-            ]}
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": user_message}
         ]
         
         # Make the API call