Update app.py

app.py

@@ -21,6 +21,9 @@ from youtube_transcript_api import YouTubeTranscriptApi
 import re
 
 
+from smolagents import Tool
+from youtube_transcript_api import YouTubeTranscriptApi
+
 from smolagents import Tool
 from youtube_transcript_api import YouTubeTranscriptApi
 
@@ -30,17 +33,17 @@ class YouTubeTranscriptTool(Tool):
     
     inputs = {
         "url": {
-            "type": "string",  # ✅ FIXED: Use OpenAPI type
+            "type": "string",
             "description": "YouTube video URL"
         }
     }
     outputs = {
         "transcript": {
-            "type": "string",  # ✅ FIXED: Use OpenAPI type
+            "type": "string",
             "description": "Transcript text from the video"
         }
     }
-    output_type = "dict"
+    output_type = "object"  # ✅ FIXED: 'object' is a valid OpenAPI type
 
     def __call__(self, url: str) -> dict:
         video_id = url.split("v=")[-1].split("&")[0]
@@ -48,9 +51,6 @@ class YouTubeTranscriptTool(Tool):
         transcript_text = " ".join([t["text"] for t in transcript_list])
         return {"transcript": transcript_text}
 
-
-
-
 transcript_tool = YouTubeTranscriptTool()
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):