Update app.py

app.py

@@ -11,6 +11,7 @@ from langchain.prompts.chat import SystemMessagePromptTemplate, ChatPromptTempla
 from langchain_huggingface import HuggingFaceEndpoint
 from huggingface_hub import login
 from langchain.llms import HuggingFaceHub
+from agent import build_agent
 
 
 
@@ -51,29 +52,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
      # llm = ChatOllama(model="llama3.1")
-        login(token=os.getenv('HF_TOKEN'))
-
-# Use a Hugging Face cloud model
-        llm = HuggingFaceHub(
-            repo_id="google/flan-t5-base",  # Example cloud model; change as needed, 
-            task="text-generation"
-        )
-
-        search = DuckDuckGoSearchRun()
-        tools = [
-              Tool(
-                  name="DuckDuckGo Search",
-                  func=search.run,
-                  description="Useful for answering questions about current events or the web."
-               )
-            ]
-        agent = initialize_agent(
-                tools,
-                llm,
-                agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-                verbose=True,
-                handle_parsing_errors=True
-           )
+        agent = build_agent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None