Update app.py

app.py

@@ -24,18 +24,50 @@ def markdown_to_html(text):
 
 # IMPORTANT: Replace with your actual Gemini API key
 # Load API Keys (Gemini)
-# Load API Keys (Gemini)
-formatted_gemini_key = os.getenv('GEMINI_API_KEY') or os.getenv('GEMINI_API_KEY_1') or os.getenv('GEMINI_API_KEY_2')
+# --- ROBUST API KEY DETECTION ---
+def get_api_key(service_type):
+    """
+    Intelligently fetches API keys by scanning multiple environment variables.
+    - service_type="gemini": Looks for keys starting with "AIza"
+    - service_type="nvidia": Looks for keys starting with "nvapi-"
+    """
+    potential_vars = [
+        "GEMINI_API_KEY", "GEMINI_API_KEY_1", "GEMINI_API_KEY_2", "GEMINI_API",
+        "NVIDIA_API_KEY", "NVIDIA_API"
+    ]
+    
+    found_key = None
+    
+    for var_name in potential_vars:
+        val = os.getenv(var_name)
+        if not val:
+            continue
+            
+        if service_type == "gemini" and val.startswith("AIza"):
+            print(f"DEBUG: Found Gemini Key in variable '{var_name}'")
+            return val
+        elif service_type == "nvidia" and val.startswith("nvapi-"):
+            print(f"DEBUG: Found NVIDIA Key in variable '{var_name}'")
+            return val
+            
+    return None
+
+# Initialize Clients
+gemini_key = get_api_key("gemini")
+nvidia_key = get_api_key("nvidia")
 
 gemini_client = None
-if formatted_gemini_key:
-    print(f"Initializing Gemini Client with API key: {formatted_gemini_key[:10]}...")
+if gemini_key:
+    print(f"Initializing Gemini Client with key: {gemini_key[:10]}...")
     try:
-        gemini_client = genai.Client(api_key=formatted_gemini_key)
+        gemini_client = genai.Client(api_key=gemini_key)
     except Exception as e:
         print(f"Error initializing Gemini Client: {e}")
 else:
-    print("WARNING: GEMINI_API_KEY not found. Gemini models will be skipped.")
+    print("WARNING: No valid Gemini API Key (starting with 'AIza') found in environment variables.")
+
+# NVIDIA fallback logic will use `nvidia_key` later
+
 def validate_coordinates(lat, lon):
     """Validate and convert latitude and longitude to float."""
     try:
@@ -425,12 +457,13 @@ def call_gemini_api(input_data, language):
         print("DEBUG: Gemini client not initialized. Skipping Gemini models.")
 
     # 4. NVIDIA Fallback (if Gemini fails)
-    if not report_data and os.getenv("NVIDIA_API_KEY"):
+    if not report_data and nvidia_key:
         try:
             from openai import OpenAI
+            # Use the detected nvidia_key
             nv_client = OpenAI(
                 base_url="https://integrate.api.nvidia.com/v1",
-                api_key=os.getenv("NVIDIA_API_KEY")
+                api_key=nvidia_key
             )
             
             # Council of models for text generation
@@ -467,6 +500,8 @@ def call_gemini_api(input_data, language):
                     continue
         except Exception as e:
             print(f"CRITICAL: NVIDIA client setup failed: {e}")
+    elif not report_data:
+        print("DEBUG: No valid report and NVIDIA key not found/valid.")
     
     if not report_data:
          print("DEBUG: All models failed to generate valid report.")