Update app.py

app.py

@@ -15,61 +15,6 @@ openai_key = os.environ.get("OPENAI_API_KEY")
 
 search_tool = DuckDuckGoSearchTool()
 
-##Tool 1
-from smolagents import Tool
-from youtube_transcript_api import YouTubeTranscriptApi
-from urllib.parse import urlparse, parse_qs
-
-from smolagents import Tool
-from youtube_transcript_api import YouTubeTranscriptApi
-from urllib.parse import urlparse, parse_qs
-
-class YouTubeDialogueTool(Tool):
-    name = "youtube_dialogue_qa"
-    description = "Extracts the transcript and finds what Teal'c says in response to a specific question."
-
-    inputs = {
-        "url": {
-            "type": "string",
-            "description": "Full YouTube video URL"
-        },
-        "question": {
-            "type": "string",
-            "description": 'Exact question asked (e.g., "Isn\'t that hot?")'
-        },
-        "speaker": {
-            "type": "string",
-            "description": "Name of the character who responds (e.g., 'Teal'c')",
-            "nullable": True
-        }
-    }
-
-    output_type = "string"
-
-    def forward(self, url: str, question: str, speaker: str = "Teal'c") -> str:
-        try:
-            parsed = urlparse(url)
-            video_id = parse_qs(parsed.query).get("v", [None])[0]
-            if not video_id:
-                return "ERROR: Invalid YouTube URL or missing video ID."
-
-            transcript = YouTubeTranscriptApi.get_transcript(video_id)
-
-            lines = [entry["text"].replace("\n", " ").strip() for entry in transcript if entry["text"].strip()]
-
-            for idx, line in enumerate(lines):
-                if question.lower().rstrip("?") in line.lower():
-                    if idx + 1 < len(lines):
-                        return lines[idx + 1].strip()
-                    break
-
-            return "No response found in transcript."
-
-        except Exception as e:
-            return f"Error retrieving transcript: {str(e)}"
-
-
-
 
 ##Tool 2
 import wikipedia
@@ -130,7 +75,7 @@ class WikipediaQATool(Tool):
 
 wiki_tool = WikipediaQATool()
 #excel_tool = ExcelAnalysisTool()
-yt_tool = YouTubeDialogueTool()
+#yt_tool = YouTubeDialogueTool()
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):
     log_output = ""
@@ -138,8 +83,8 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
     try:
         
         agent = ToolCallingAgent(
-            tools=[search_tool, yt_tool],
-                model=OpenAIServerModel(model_id="gpt-4o-mini", 
+            tools=[search_tool],
+                model=OpenAIServerModel(model_id="gpt-4o", 
                                         api_key=os.environ["OPENAI_API_KEY"], 
                                        temperature=0.0),
             max_steps=7,