Update app.py

app.py

@@ -5,6 +5,29 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 from google import genai
+from dotenv import load_dotenv
+load_dotenv()
+gemini_api_key = os.getenv("GEMINI_API_KEY")
+if not gemini_api_key:
+    # Option 1: Raise an error if the key is absolutely required
+    raise ValueError("GEMINI_API_KEY environment variable not set. Please set it in Spaces Secrets.")
+    # Option 2: Handle gracefully (e.g., disable features, show a message)
+    # print("Warning: GEMINI_API_KEY not found. Gemini features will be disabled.")
+    # gemini_features_enabled = False
+else:
+    # print("Gemini API Key found.") # Avoid printing the key itself!
+    # gemini_features_enabled = True
+    # --- Configure the Gemini client ---
+    try:
+        genai.configure(api_key=gemini_api_key)
+        print("Gemini client configured successfully.")
+        # You can now initialize your model, e.g.:
+        # model = genai.GenerativeModel('gemini-pro')
+        # response = model.generate_content("Tell me a joke.")
+        # print(response.text)
+    except Exception as e:
+         print(f"Error configuring Gemini client: {e}")
+        
 
 # Import AgentText to return a composite Markdown message
 from smolagents.agent_types import AgentText
@@ -58,17 +81,11 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
         messages.append(f"Note: {arg1}")
     
     return "\n".join(messages)
-@tool
+  
 
 
 
 
-response = client.models.generate_content(
-    model="gemini-2.0-flash",
-    config=types.GenerateContentConfig(
-        system_instruction="You are a cat. Your name is Neko."),
-    contents="Hello there"
-)
 
 print(response.text)