Update app.py

app.py

@@ -57,6 +57,19 @@ except ImportError as e:
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(module)s - %(message)s')
 
+# 1. Set Vertex AI usage preference (if applicable)
+os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "False" 
+
+# 2. Get your API key from your chosen environment variable name
+#    Replace "YOUR_GEMINI_API_KEY_ENV_NAME" with the actual name you use, e.g., "GEMINI_API_KEY" or "GOOGLE_API_KEY"
+user_provided_api_key = os.environ.get("GEMINI_API_KEY") 
+
+if user_provided_api_key:
+    os.environ["GOOGLE_API_KEY"] = user_provided_api_key
+    logging.info("GOOGLE_API_KEY environment variable has been set from YOUR_GEMINI_API_KEY_ENV_NAME.")
+else:
+    logging.error(f"CRITICAL ERROR: The API key environment variable 'YOUR_GEMINI_API_KEY_ENV_NAME' was not found. The application may not function correctly.")
+
 
 # --- Gradio UI Blocks ---
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),