Update nova_agent.py

nova_agent.py

@@ -2,7 +2,6 @@ import os
 import boto3
 import json
 from dotenv import load_dotenv
-from video_parser import VideoParser
 from excel_parser import ExcelParser
 import re
 
@@ -34,7 +33,6 @@ class NovaProAgent:
         self.accept = "application/json"
         
         # Initialize parsers
-        self.video_parser = VideoParser()
         self.excel_parser = ExcelParser()
         
     async def __call__(self, question: str) -> str:
@@ -65,17 +63,40 @@ class NovaProAgent:
         
         url = youtube_url.group()
         
+        # Extract video ID for reference
+        video_id = re.search(r'v=([\w-]+)', url).group(1)
+        
+        # Use Nova Pro to provide intelligent response about video analysis
+        video_prompt = f"""User is asking about a YouTube video: {url}
+Video ID: {video_id}
+User question: {question}
+
+Provide a helpful response about video analysis limitations and suggest alternatives."""
+        
+        payload = {
+            "messages": [{
+                "role": "user",
+                "content": [{"text": video_prompt}]
+            }],
+            "inferenceConfig": {
+                "max_new_tokens": 150,
+                "temperature": 0.0
+            }
+        }
+        
         try:
-            # Extract video metadata without downloading
-            metadata = self.video_parser.get_youtube_metadata(url)
-            
-            if 'error' in metadata:
-                return f"Video analysis failed: {metadata['error']}"
+            response = self.bedrock_client.invoke_model(
+                modelId=self.model_id,
+                contentType=self.content_type,
+                accept=self.accept,
+                body=json.dumps(payload)
+            )
             
-            return f"Video: {metadata['title']} | Duration: {metadata['duration']}s | Views: {metadata['view_count']:,}"
+            response_body = json.loads(response['body'].read())
+            return response_body['output']['message']['content'][0]['text'].strip()
             
         except Exception as e:
-            return f"Video analysis failed: {str(e)}"
+            return f"Video ID: {video_id}. Direct video analysis unavailable due to access restrictions."
     
     async def _handle_excel_question(self, question: str) -> str:
         """Handle questions that require Excel file analysis"""