Update app.py

app.py

@@ -4,11 +4,11 @@ import io
 from PIL import Image
 from together import Together
 
-# Initialize the Together AI client
-client = Together()
-
-def generate_image(prompt):
+def generate_image(api_key, prompt):
     try:
+        # Initialize the Together AI client with the provided API key
+        client = Together(api_key=api_key)
+        
         # Call the Together API to generate an image
         response = client.images.generate(
             prompt=prompt,
@@ -38,6 +38,11 @@ with gr.Blocks() as app:
     
     with gr.Row():
         with gr.Column():
+            api_key_input = gr.Textbox(
+                label="Together API Key",
+                placeholder="Enter your Together API key here...",
+                type="password"  # Mask the API key for security
+            )
             prompt_input = gr.Textbox(
                 label="Enter your prompt",
                 placeholder="A beautiful sunset over mountains...",
@@ -50,18 +55,18 @@ with gr.Blocks() as app:
     
     generate_button.click(
         fn=generate_image,
-        inputs=prompt_input,
+        inputs=[api_key_input, prompt_input],
         outputs=image_output
     )
     
     gr.Markdown("""
     ## Instructions
-    1. Enter a descriptive prompt in the text box
-    2. Click "Generate Image" 
-    3. Wait for the image to be generated
+    1. Enter your Together AI API key (get one from https://www.together.ai)
+    2. Enter a descriptive prompt in the text box
+    3. Click "Generate Image" 
+    4. Wait for the image to be generated
     
-    Note: You'll need to have your Together AI API key set up properly, 
-    usually through environment variables or a config file.
+    Note: Your API key is not stored and is only used for the current session.
     """)
 
 # Launch the app