Update app.py

app.py

@@ -36,6 +36,9 @@ def generate_gradio_app(api_key, image):
     if not api_key:
         return "Error: API key not provided. Please enter your Together API key."
 
+    if image is None:
+        return "Error: No image provided. Please upload an image."
+
     try:
         initialize_client(api_key)
 
@@ -53,14 +56,15 @@ def generate_gradio_app(api_key, image):
 8. Provide a runnable Gradio application focusing on the most important aspects of the UI.
 9. Keep the code concise, aiming for no more than 2000 tokens.
 
-Please generate the Gradio code based on the provided image, focusing on the most crucial elements to fit within the token limit.
-
-[IMAGE]data:image/jpeg;base64,{encoded_image}[/IMAGE]"""
+Please generate the Gradio code based on the provided image, focusing on the most crucial elements to fit within the token limit."""
 
         messages = [
             {
                 "role": "user",
-                "content": prompt
+                "content": [
+                    {"type": "text", "text": prompt},
+                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}"}}
+                ]
             }
         ]