Update models/gpt4o.py

models/gpt4o.py

@@ -83,8 +83,11 @@ def run_model(image: Image.Image, api_key: str = None):
     prompt_text = load_prompt()
     encoded_image = encode_image_base64(image)
 
+    # Pull from environment if not passed explicitly
+    api_key = api_key or os.getenv("OPENAI_API_KEY")
+
     if not api_key:
-        return {"json": None, "raw": "⚠️ API key is missing. Please provide your OpenAI API key."}
+        return {"json": None, "raw": "⚠️ API key is missing. Please set it as a secret in your Space or upload it as a file."}
 
     client = openai.OpenAI(api_key=api_key)
     response = client.chat.completions.create(
@@ -108,4 +111,4 @@ def run_model(image: Image.Image, api_key: str = None):
     full_ocr_text = extract_all_text_pix2struct(image)
     parsed_json = assign_event_gateway_names_from_ocr(image, parsed_json, full_ocr_text)
 
-    return {"json": parsed_json, "raw": output_text}
+    return {"json": parsed_json, "raw": output_text}