Update crew.py

crew.py

@@ -153,6 +153,32 @@ def run_crew(question, file_path):
             return response.text
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
+            
+    @tool("YouTube Analysis Tool")
+    def youtube_analysis_tool(question: str, url: str) -> str:
+        """Given a question and YouTube URL, analyze the video to answer the question.
+    
+           Args:
+               question (str): Question about a YouTube video
+               url (str): The YouTube video URL
+                
+           Returns:
+               str: Answer to the question about the YouTube video
+                
+           Raises:
+               RuntimeError: If processing fails"""
+        try:
+            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+
+            return client.models.generate_content(
+                model=YOUTUBE_ANALYSIS_MODEL,
+                contents=types.Content(
+                    parts=[types.Part(file_data=types.FileData(file_uri=url)),
+                           types.Part(text=question)]
+                )
+            )
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Document Analysis Tool")
     def document_analysis_tool(question: str, file_path: str) -> str:
@@ -172,7 +198,7 @@ def run_crew(question, file_path):
 
             contents = []
             
-            if is_ext(file_path, ".docx")
+            if is_ext(file_path, ".docx"):
                 file_data = read_docx(file_path)
                 contents = [f"{question}\n{file_data}"]
             else:
@@ -187,33 +213,7 @@ def run_crew(question, file_path):
             return response.text
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
-            
-    @tool("YouTube Analysis Tool")
-    def youtube_analysis_tool(question: str, url: str) -> str:
-        """Given a question and YouTube URL, analyze the video to answer the question.
-    
-           Args:
-               question (str): Question about a YouTube video
-               url (str): The YouTube video URL
-                
-           Returns:
-               str: Answer to the question about the YouTube video
-                
-           Raises:
-               RuntimeError: If processing fails"""
-        try:
-            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
-            return client.models.generate_content(
-                model=YOUTUBE_ANALYSIS_MODEL,
-                contents=types.Content(
-                    parts=[types.Part(file_data=types.FileData(file_uri=url)),
-                           types.Part(text=question)]
-                )
-            )
-        except Exception as e:
-            raise RuntimeError(f"Processing failed: {str(e)}")
-    
     @tool("Code Generation Tool")
     def code_generation_tool(question: str, file_path: str) -> str:
         """Given a question and data file, generate and execute code to answer the question.