Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import os
 from together import Together
 import base64
 from io import BytesIO
@@ -7,15 +6,14 @@ from PIL import Image
 import numpy as np
 import traceback
 
-# Initialize the Together client
-api_key = os.environ.get('TOGETHER_API_KEY')
-client = Together(api_key=api_key)
-
-def generate_gradio_app(image):
+def generate_gradio_app(api_key, image):
     if not api_key:
-        return "Error: TOGETHER_API_KEY not set. Please check your API key."
+        return "Error: API key not provided. Please enter your Together API key."
 
     try:
+        # Initialize the Together client with the provided API key
+        client = Together(api_key=api_key)
+
         # Convert numpy array to PIL Image
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image.astype('uint8'), 'RGB')
@@ -82,11 +80,13 @@ Please generate the entire Gradio code based on the provided image."""
     except Exception as e:
         error_message = str(e)
         stack_trace = traceback.format_exc()
-        return f"An error occurred: {error_message}\n\nStack trace:\n{stack_trace}\n\nPlease try again or check your API key and connection."
+        return f"An error occurred: {error_message}\n\nStack trace:\n{stack_trace}\n\nPlease check your API key and try again."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Generate Gradio App from Wireframe")
-    gr.Markdown("Upload an image of your UI design, and we'll generate Gradio code to recreate it.")
+    gr.Markdown("Enter your Together API key, upload an image of your UI design, and we'll generate Gradio code to recreate it.")
+    
+    api_key_input = gr.Textbox(label="Enter your Together API Key", type="password")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
     
     generate_button.click(
         fn=generate_gradio_app,
-        inputs=[image_input],
+        inputs=[api_key_input, image_input],
         outputs=[code_output]
     )