Update app.py

app.py

@@ -40,21 +40,13 @@ def generate_gradio_app(api_key, image):
 
 Please generate the entire Gradio code based on the provided image."""
 
-        # Make the API call
+        # Make the API call with the corrected message format
         response = client.chat.completions.create(
             model="meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
             messages=[
                 {
                     "role": "user",
-                    "content": [
-                        {"type": "text", "text": prompt},
-                        {
-                            "type": "image_url",
-                            "image_url": {
-                                "url": f"data:image/png;base64,{img_str}",
-                            },
-                        },
-                    ],
+                    "content": prompt + f"\n\n<image>\ndata:image/png;base64,{img_str}\n</image>"
                 }
             ],
             max_tokens=4096,
@@ -63,7 +55,7 @@ Please generate the entire Gradio code based on the provided image."""
             top_k=50,
             repetition_penalty=1,
             stop=["<|eot_id|>", "<|eom_id|>"],
-            stream=False  # Changed to False for easier debugging
+            stream=False
         )
         
         # Debug: Print the entire response