Update app.py

app.py

@@ -1,3 +1,22 @@
+"""
+This is an example template to build a chatbox app using agentic AI
+
+Given a question, this agent will:
++ Derive websearch query
++ Retrieve relevant information
++ Interprete the information and return the answer
+
+To make the agent more capable, we can add other tools in the future, for example, CSV loading, calculations
+
+Agentic framework used: Smolagents (HuggingFace)
+LLM: gemini/gemini-2.0-flash-lite
+
+@Contributor: nhatquynhthuyen.truong@cotiviti.com
+@Org: Cotiviti AU
+@Release Date: 28 Aug 2025
+@Last Update: 28 Aug 2025
+"""
+
 import os
 os.system("pip install -q smolagents ddgs litellm markdownify requests")
 os.system("pip install -q llm_rs")
@@ -5,55 +24,48 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, LiteLLMModel, InferenceClientModel, VisitWebpageTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, LiteLLMModel, VisitWebpageTool
 import time
 import random
-import time
-import gradio as gr
-from llm_rs import AutoModel
 import logging
 import numpy as np
 from typing import List
-import requests
 import markdownify
 from smolagents import tool
 
+
 api_key = os.environ.get('Google_api_key')
 llm = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=api_key)
 SYS_PROMPT = """You are an assistant for answering questions.
 If you don't know the answer, it's OK to make a guess."""
 
+
 # --- Basic Agent Definition ---
-class BasicAgent:  
+class BasicAgent:
     def __init__(self):
-        self.agent =  CodeAgent(tools=[DuckDuckGoSearchTool(),VisitWebpageTool()
+        self.agent = CodeAgent(tools=[DuckDuckGoSearchTool(), VisitWebpageTool()
                                       ], model=llm)
         print("BasicAgent initialized.")
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         final_answer = self.agent.run(question)
         print(f"Agent returning final answer: {final_answer}")
         return final_answer
 
+
 def run_and_submit_all(question_text):
     """
     Fetches the question, runs the BasicAgent on it and return answer
     """
-    # --- Determine HF Space Runtime URL and Repo URL ---
-    space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
-
-
-    # 1. Instantiate Agent ( modify this part to create your agent)
+    # 1. Instantiate Agent
     try:
         agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
-    # In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
-    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
-    print(agent_code)
-    
-    # 2. Run your Agent
+
+    # 2. Run Agent
     try:
         agent_answer = agent(question_text)
     except Exception as e:
@@ -80,25 +92,6 @@ demo.launch()
 
 
 if __name__ == "__main__":
-    print("\n" + "-"*30 + " App Starting " + "-"*30)
-    # Check for SPACE_HOST and SPACE_ID at startup for information
-    space_host_startup = os.getenv("SPACE_HOST")
-    space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
-
-    if space_host_startup:
-        print(f"✅ SPACE_HOST found: {space_host_startup}")
-        print(f"   Runtime URL should be: https://{space_host_startup}.hf.space")
-    else:
-        print("ℹ️  SPACE_HOST environment variable not found (running locally?).")
-
-    if space_id_startup: # Print repo URLs if SPACE_ID is found
-        print(f"✅ SPACE_ID found: {space_id_startup}")
-        print(f"   Repo URL: https://huggingface.co/spaces/{space_id_startup}")
-        print(f"   Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
-    else:
-        print("ℹ️  SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
-
-    print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)