Update app.py

app.py

@@ -4,21 +4,21 @@ import os
 from together import Together
 import base64
 import io
-from dotenv import load_dotenv
-
-# Load environment variables from .env file
-load_dotenv()
 
 # Initialize Together client
 client = None
-api_key=None
+
 def initialize_client(api_key=None):
     global client
-    # Load the API key from environment if not provided
-    api_key = os.getenv("TG_API_KEY")
+    # Fetch the API key from the environment if it's not provided directly
+    api_key = api_key or os.getenv("TOGETHER_API_KEY")
     print(api_key)
-    client = Together()
-    
+    if api_key:
+        os.environ["TOGETHER_API_KEY"] = api_key
+        client = Together()
+    else:
+        raise ValueError("Please provide a Together API Key")
+
 
 def encode_image(image_path):
     with Image.open(image_path) as img:
@@ -26,15 +26,13 @@ def encode_image(image_path):
         img.save(buffered, format="PNG")
         return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-def bot_streaming(
-    message, history
-):
-    together_api_key=api_key
-    max_new_tokens=250
+
+def bot_streaming(message, history):
+    max_new_tokens=250 
     temperature=0.7
     if client is None:
         try:
-            initialize_client(together_api_key)
+            initialize_client()
         except Exception as e:
             history.append((message, f"Error initializing client: {str(e)}"))
             yield history
@@ -133,34 +131,15 @@ def bot_streaming(
 
 # The rest of your Gradio interface code remains the same
 with gr.Blocks() as demo:
-    gr.Markdown("# Meta Llama-3.2-11B-Vision-Instruct (FREE)")
+    gr.Markdown("# Medical Image Analyzer")
     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 and get started!"
+        "Upload a medical image, and start chatting about it."
     )
 
-    with gr.Row():
-        together_api_key = gr.Textbox(
-            label="Together API Key",
-            placeholder="Enter your TOGETHER_API_KEY here",
-            type="password",
-        )
-
-    with gr.Row():
-        # max_new_tokens = gr.Slider(
-        #     minimum=10,
-        #     maximum=10000,
-        #     value=250,
-        #     step=10,
-        #     label="Maximum number of new tokens",
-        # )
-        # temperature = gr.Number(
-        #     value=0.7, minimum=0, maximum=1, step=0.1, label="Temperature"
-        # )
-
     chatbot = gr.Chatbot()
     msg = gr.MultimodalTextbox(label="")
     clear = gr.Button("Clear")
-    
+
     msg.submit(
         bot_streaming,
         [msg, chatbot],