Add key

src/app.py

@@ -6,11 +6,14 @@ import os
 from datasets import load_dataset
 from PIL import Image
 import io
+from dotenv import load_dotenv
 
 from src.modules.vlm_inference import analyze_product_image
 from src.modules.data_processing import pil_to_base64
 from src.modules.evals import run_inference_on_dataframe
 
+load_dotenv()
+
 # Constants
 AVAILABLE_MODELS = {
     "Qwen3-VL-30B-A3B": "accounts/fireworks/models/qwen3-vl-30b-a3b-instruct",
@@ -35,7 +38,6 @@ PROMPT_STYLES = {
 def analyze_single_image(
     image_input,
     model_name: str,
-    api_key: Optional[str] = None,
     prompt_style_display: Optional[str] = None,
 ) -> tuple[str, str, str, str]:
     """
@@ -59,11 +61,7 @@ def analyze_single_image(
 
         # Determine provider from model name
         model_id = AVAILABLE_MODELS[model_name]
-
-        # Get API key from environment if not provided
-        if api_key is None:
-            api_key = os.getenv("FIREWORKS_API_KEY")
-
+        api_key = os.getenv("FIREWORKS_API_KEY")
         # Map display name to prompt key
         prompt_style = (
             PROMPT_STYLES.get(prompt_style_display) if prompt_style_display else None
@@ -243,10 +241,6 @@ def create_demo_interface():
                 value="Website/Sales",
                 label="Description Style",
             )
-            api_key_input = gr.Textbox(
-                label="API Key",
-                type="password",
-            )
 
         with gr.Tabs():
             with gr.TabItem("📸 Image Analysis 📸 "):
@@ -293,7 +287,6 @@ def create_demo_interface():
                     inputs=[
                         image_input,
                         model_selector,
-                        api_key_input,
                         prompt_selector,
                     ],
                     outputs=[