Update app.py

app.py

@@ -14,11 +14,11 @@ client = None
 def initialize_client(api_key=None):
     global client
     if api_key:
-        client = Together(api_key=api_key)
-    elif "TOGETHER_API_KEY" in os.environ:
+        os.environ["TOGETHER_API_KEY"] = api_key
+    if "TOGETHER_API_KEY" in os.environ:
         client = Together()
     else:
-        raise ValueError("Please provide an API key or set the TOGETHER_API_KEY environment variable")
+        raise ValueError("Please provide a Together API Key")
 
 def encode_image(image_path, max_size=(800, 800), quality=85):
     with Image.open(image_path) as img:
@@ -31,9 +31,9 @@ def encode_image(image_path, max_size=(800, 800), quality=85):
         img.save(buffered, format="JPEG", quality=quality)
         return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
-def bot_streaming(message, history, api_key, max_new_tokens=250, max_history=5):
+def bot_streaming(message, history, together_api_key, max_new_tokens=250, max_history=5):
     if client is None:
-        initialize_client(api_key)
+        initialize_client(together_api_key)
 
     txt = message["text"]
     messages = []
@@ -83,9 +83,9 @@ with gr.Blocks() as demo:
     gr.Markdown("# Meta Llama-3.2-11B-Vision-Instruct (FREE)")
     gr.Markdown("Try the new Llama 3.2 11B Vision API by Meta for free through Together AI. Upload an image, and start chatting about it. Just paste in your [Together AI API key](https://api.together.xyz/settings/api-keys) and get started!")
     
-    api_key = gr.Textbox(
+    together_api_key = gr.Textbox(
         label="Together API Key",
-        placeholder="Enter your API key here.",
+        placeholder="Enter your TOGETHER_API_KEY here",
         type="password"
     )
     
@@ -107,7 +107,7 @@ with gr.Blocks() as demo:
         multimodal=True
     )
     
-    api_key.change(lambda x: x, inputs=[api_key], outputs=[chatbot.additional_inputs[0]])
+    together_api_key.change(lambda x: x, inputs=[together_api_key], outputs=[chatbot.additional_inputs[0]])
 
 if __name__ == "__main__":
     demo.launch(debug=True)