Update app.py

app.py

@@ -60,12 +60,12 @@ def generate_gradio_app(image):
         response = client.chat.completions.create(
             model="meta-llama/Llama-Vision-Free",
             messages=messages,
-            max_tokens=50000,
+            max_tokens=2048,
             temperature=0.7,
             top_p=0.7,
             top_k=50,
             repetition_penalty=1,
-            stop=["<|eot_id|>", "<|eom_id|>"],
+            stop=["<|im_end|>"],
             stream=True
         )
         
@@ -77,26 +77,19 @@ def generate_gradio_app(image):
         
         return generated_code
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return f"An error occurred: {str(e)}\n\nPlease try again or check your API key and connection."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Turn your wireframe into a Gradio app")
     gr.Markdown("Upload an image of your UI design and we'll build a Gradio app for you.")
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             image_input = gr.Image(label="Upload a screenshot", elem_id="image_upload")
-            example_link = gr.Markdown("Need an example image? [Try ours](https://example.com/wireframe.png).")
-            
-            model_dropdown = gr.Dropdown(
-                choices=["Llama-Vision-Free"],
-                value="Llama-Vision-Free",
-                label="AI Model"
-            )
-            
             generate_button = gr.Button("Generate Gradio app", variant="primary")
-    
-    code_output = gr.Code(language="python", label="Generated Gradio Code", lines=30)
+        
+        with gr.Column(scale=2):
+            code_output = gr.Code(language="python", label="Generated Gradio Code", lines=30)
     
     generate_button.click(
         fn=generate_gradio_app,