added gemini api requirement.

app.py

@@ -28,9 +28,10 @@ BLAXEL_ICON_URL = "https://huggingface.co/datasets/DEVAIEXP/assets/resolve/main/
 MCP_SERVER_URL = os.getenv("MCP_SERVER_URL", "https://mcp-1st-birthday-gitrepo-inspector-mcp.hf.space/")
 AGENT_API_URL = os.getenv("AGENT_API_URL", "https://run.blaxel.ai/devaiexp/agents/agent")
 BLAXEL_API_KEY = os.getenv("BLAXEL_API_KEY")
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
 # LOGIC LAYER
-def _validate_api_keys(llm_api_key, request):
+def _validate_api_keys(llm_api_key, request, type="LLM"):
     USE_SERVER_KEYS = os.getenv("USE_SERVER_KEYS", "false").lower() in (
         "True",
         "true",
@@ -39,7 +40,7 @@ def _validate_api_keys(llm_api_key, request):
     )
     if not USE_SERVER_KEYS and request and request.headers.get("referer"):
         if not llm_api_key or not llm_api_key.strip():
-            raise gr.Error("⚠️ LLM API Key Required! Please enter your own API Key to use this tool in the demo UI.")
+            raise gr.Error(f"⚠️ {type} API Key Required! Please enter your own API Key to use this tool in the demo UI.")
 
 def _get_custom_header():
     html = """
@@ -517,8 +518,8 @@ def generate_priority_report(repo_url, provider, model, api_key, request: gr.Req
     Generates priority report using Sidebar configs.
     Handles tuple return (html, thought) from MCP.
     """
+    _validate_api_keys(api_key, request)
     gr.Info(f"🧠 Generating Strategy Report for {repo_url}... Please wait.")
-    llm_api_key = _validate_api_keys(api_key, request)
 
     try:
         client = Client(MCP_SERVER_URL)        
@@ -526,7 +527,7 @@ def generate_priority_report(repo_url, provider, model, api_key, request: gr.Req
             repo_url,
             provider, 
             model, 
-            llm_api_key,
+            api_key,
             api_name="/prioritize_open_issues"
         )
                 
@@ -708,7 +709,12 @@ with gr.Blocks(title="GitRepo Inspector", theme=theme, css=css_code, head=APP_HE
             placeholder="your API Key...",
             info="Required for approval and reanalysis actions"
         )
-
+        global_gemini_key = gr.Textbox(
+            label="Gemini Assistant API Key", 
+            type="password", 
+            placeholder="your API Key...",
+            info="Gemini API Key required for AI Assistant Chat"
+        )
         gr.Markdown("---")
         log_limit_slider = gr.Slider(
             minimum=10, 
@@ -779,14 +785,16 @@ with gr.Blocks(title="GitRepo Inspector", theme=theme, css=css_code, head=APP_HE
                 inputs=None,
                 outputs=[chatbot]
             )
-            def init_agent():
-                return create_dashboard_agent()
+            def init_agent(gemini_api_key):                
+                return create_dashboard_agent(gemini_api_key)
   
          
-            def interact(agent, prompt, history, current_repo, provider, model, token):                
+            def interact(agent, prompt, history, current_repo, provider, model, token, gemini_api_key, request: gr.Request=None):
                 agent_state = agent
                 if agent_state is None:
-                    agent_state = init_agent()
+                    _validate_api_keys(gemini_api_key, request, "Gemini Assistant")
+                    gr.Info("⚡ Please wait, initializing the agent for the first time...")
+                    agent_state = init_agent(gemini_api_key)
                     if agent_state is None:
                         history.append(gr.ChatMessage(role="user", content=prompt, metadata={"status": "done"}))
                         history.append(gr.ChatMessage(
@@ -864,7 +872,8 @@ with gr.Blocks(title="GitRepo Inspector", theme=theme, css=css_code, head=APP_HE
                     global_repo_url,
                     global_provider,
                     global_model,
-                    global_github_token 
+                    global_github_token,
+                    global_gemini_key
                 ], 
                 outputs=[chatbot, msg, chat_trigger_issue, agent_state],
                 show_progress="hidden"
@@ -1025,10 +1034,10 @@ with gr.Blocks(title="GitRepo Inspector", theme=theme, css=css_code, head=APP_HE
     ]
     
     # Added log_limit_slider to inputs
-    app.load(
-         init_agent, 
-         outputs=[agent_state]
-    )
+    # app.load(
+    #      init_agent, 
+    #      outputs=[agent_state]
+    # )
     app.load(
         fn=hard_refresh,
         inputs=[view_filter, current_sort, log_limit_slider, global_repo_url],

services/agent_chat.py

@@ -3,7 +3,7 @@ import os
 import re
 import json
 from typing import Dict, List, Any, Optional
-from smolagents import CodeAgent, LiteLLMModel, tool, ToolCallingAgent
+from smolagents import LiteLLMModel, tool, ToolCallingAgent
 from smolagents.mcp_client import MCPClient as SmolMCPClient
 from gradio_client import Client as GradioClient
 from config.constants import AVAILABLE_MODELS_BY_PROVIDER
@@ -176,13 +176,14 @@ def trigger_live_analysis(
 #                         AGENT FACTORY
 # ===================================================================
 
-def create_dashboard_agent():
+def create_dashboard_agent(gemini_api_key):
     print("Initializing GitRepo Inspector Agent...")
 
     model = LiteLLMModel(
         model_id="gemini/gemini-2.5-flash",
         temperature=0.1,
-        max_tokens=2048
+        max_tokens=2048,
+        api_key=gemini_api_key
     )
 
     # Loads the clean prompt from YAML config