Update app.py

app.py

@@ -12,6 +12,7 @@ from langgraph.graph import StateGraph, END
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
 from langchain_community.document_loaders import WikipediaLoader
+from youtube_transcript_api import YouTubeTranscriptApi
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -123,6 +124,31 @@ def wikipedia_tool(query: str) -> str:
     except Exception as e:
         return f"Wikipedia search error: {e}"
     
+@tool
+def youtube_transcript_tool(video_id: str,query:str) -> str:
+    """
+    Fetches the transcript of a YouTube video.
+
+    Args:
+        video_id (str): The YouTube video ID.
+        query (str): The question to be answered based on the transcript.
+
+    Returns:
+        str: The transcript text or an error message.
+    """
+    try:
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        question = f"Answer the question based on the transcript: {query}"
+        prompt = (
+            f"Given the following YouTube transcript, answer the question as directly as possible:\n"
+            f"Question: {question}\n"
+            f"Transcript: {transcript}\n"
+            f"Answer:"
+        )
+        answer = openrouter_inference(prompt)
+    except Exception as e:
+        return f"Transcript error: {e}"
+    
 
     
     
@@ -135,11 +161,23 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         token=os.environ["OPENROUTER_API_KEY"]  
-        self.system_prompt= """You are a helpful assistant tasked with answering questions using a set of tools. 
-Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
-FINAL ANSWER: [YOUR FINAL ANSWER]. 
-YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
-Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
+        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. 
+        Always return a valid answer after validating the source.
+        
+        Your final answer should be in the following format:
+        "your answer here".
+        
+        If you are asked question related to numbers,return the answer in numeric format.
+        If you want more details about a topic,person, you can use the wikipedia tool.
+        If you want to perform web search to find the answer, use the web_search tool.
+        If you are given a text which is not known try to rearrange the words and characters if needed to form a valid answer.
+        If you are asked to find a specific person form a work,search the work using the web_search_tool or wikipedia_tool and find the person and return the person's name.
+        If you are given an youtube video then use the youtube_transcript_tool to get the transcript of the video and then answer the question based on the transcript.
+        If you are given a task to create an image, you can use the image_generation_tool.
+        
+        Answer only what is asked, do not add any additional information.
         
         """
         
@@ -149,7 +187,7 @@ Your answer should only start with "FINAL ANSWER: ", then follows with the answe
             provider="auto"  # Use openrouter as the provider
 )
         self.agent= CodeAgent(
-            tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
+            tools = [add, subtract, multiply, divide, web_search, image_generation_tool,youtube_transcript_tool, wikipedia_tool],
             model=model,
         )
     def __call__(self, question: str, context: str = "") -> str: