Update app.py

app.py

@@ -3,21 +3,156 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+#mode requirement
+from smolagents import DuckDuckGoSearchTool
+from typing import TypedDict, List, Dict, Any, Optional
+from langgraph.graph import StateGraph, END
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import HumanMessage
+from langchain.tools import DuckDuckGoSearchRun,load_tool, Tool
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+@tool
+def add(a:int,b:int)->int:
+    """
+    Adds two integers.
+    Args:
+        a (int): The first integer.
+        b (int): The second integer.
+    Returns:
+        int: The sum of the two integers.   
+    """
+    return a + b
+
+@tool
+def subtract(a:int,b:int)->int:
+    """
+    Subtracts two integers.
+    Args:
+        a (int): The first integer.
+        b (int): The second integer.
+    Returns:
+        int: The difference of the two integers.   
+    """
+    return a - b
+
+@tool
+def multiply(a:int,b:int)->int:
+    """
+    Multiplies two integers.
+    Args:
+        a (int): The first integer.
+        b (int): The second integer.
+    Returns:
+        int: The product of the two integers.   
+    """
+    return a * b
+
+@tool
+def divide(a:int,b:int)->float:
+    """
+    Divides two integers.
+    Args:
+        a (int): The numerator.
+        b (int): The denominator.
+    Returns:
+        float: The quotient of the two integers.   
+    """
+    if b == 0:
+        raise ValueError("Division by zero is not allowed.")
+    return a / b
+
+search_tool = DuckDuckGoSearchTool()
+
+@tool
+def web_search(query: str) -> str:
+    """
+    Performs a web search for the given query.
+    Args:
+        query (str): The search query.
+    Returns:
+        str: The search results as a string.
+    """
+    result=search_tool(query)
+    return f"Search results for '{query}' : {result}."
+
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
+        os.environ["OPENAI_API_KEY"] = "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" 
+        self.system_prompt= """You are a helpful assistant. You will answer questions based on the provided context.You will always return a valid answer, even if the question is not clear or the context is insufficient. If you cannot answer, return a default answer.Always return a valid answer after validating the source.
+        Always return the answer in the following format:
+        "ANSWER: <your answer here>".
+        If the question is not clear or the context is insufficient, ask for clarification.
+        Incase of numerical questions, always return the answer in the following format:
+        "ANSWER: <your answer here> (e.g. 42, 3.14, etc.)".
+        If the question is about a specific topic, provide a brief summary of the topic.
+        If the question is about a specific person, provide a brief summary of the person's background and achievements.
+        If the question is about a specific event, provide a brief summary of the event.
+        If the question is about a specific place, provide a brief summary of the place's history and significance.
+        If the question is about a specific concept, provide a brief summary of the concept.
+        If the question is about a specific term, provide a brief definition of the term.
+        If the question is about a specific date, provide a brief summary of the significance of that date.
+        If the question is about a specific number, provide a brief summary of the significance of that number.
+        If the question is about a specific unit, provide a brief summary of the significance of that unit.
+        If the question is about a specific formula, provide a brief summary of the significance of that formula.
+        If the question is about a specific algorithm, provide a brief summary of the significance of that algorithm.
+        If the question is about a specific programming language, provide a brief summary of the significance of that programming language.
+        If the question is about a specific technology, provide a brief summary of the significance of that technology.
+        If the question is about a specific framework, provide a brief summary of the significance of that framework.
+        If the question is about a specific library, provide a brief summary of the significance of that library.
+        If the question is about a specific tool, provide a brief summary of the significance of that tool.
+        If the question is about a specific method, provide a brief summary of the significance of that method.
+        If the question is about a specific technique, provide a brief summary of the significance of that technique.
+        If the question is about a specific process, provide a brief summary of the significance of that process.
+        If the question is about a specific system, provide a brief summary of the significance of that system.
+        If the question is about a specific model, provide a brief summary of the significance of that model.
+        If the question is about a specific theory, provide a brief summary of the significance of that theory.
+        If the question is about a specific principle, provide a brief summary of the significance of that principle.
+        If the question is about a specific law, provide a brief summary of the significance of that law.
+        If the question is about a specific regulation, provide a brief summary of the significance of that regulation.
+        If the question is about a specific standard, provide a brief summary of the significance of that standard.
+        If the question is about a specific guideline, provide a brief summary of the significance of that guideline.
+        If the question is about a specific best practice, provide a brief summary of the significance of that best practice.
+        If the question is about a specific case study, provide a brief summary of the significance of that case study.
+        If the question is about a specific example, provide a brief summary of the significance of that example.
+        If the question is about a specific application, provide a brief summary of the significance of that application.
+        If the question is about a specific use case, provide a brief summary of the significance of that use case.
+        If the question is about a specific scenario, provide a brief summary of the significance of that scenario.
+        If the question is about a specific challenge, provide a brief summary of the significance of that challenge.
+        If the question is about a specific opportunity, provide a brief summary of the significance of that opportunity.
+        If the question is about a specific trend, provide a brief summary of the significance of that trend.
+        If the question is about a specific issue, provide a brief summary of the significance of that issue.
+        If the question is about a specific problem, provide a brief summary of the significance of that problem.
+        If the question is about a specific solution, provide a brief summary of the significance of that solution.
+        If the question is about a specific strategy, provide a brief summary of the significance of that strategy.
+        If the question is about a specific tactic, provide a brief summary of the significance of that tactic.
+        If the question is about a specific approach, provide a brief summary of the significance of that approach.
+        If the question is about a specific method, provide a brief summary of the significance of that method.
+        If the question is about a specific technique, provide a brief summary of the significance of that technique.
+        If you are not able to find the answer using the tools privided, you can use the web_search tool.
+        If you are given a task to create an image,you can use the image_generation_tool.
+        """
+        llm = ChatOpenAI(model="gpt-4o", temperature=0)
+        self.agent= initialize_agent(
+            tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
+            llm=llm,
+        )
+    def __call__(self, question: str,context: str="") -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        # Inject system prompt + question
+        question_with_prompt = f"{self.system_prompt}\n\nContext: {context}\n\nQuestion: {question.strip()}"
+        answer = self.agent.run(question_with_prompt)
+        print(f"Agent returning answer: {answer.strip()}")
+        return answer.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """