Update app.py

app.py

@@ -33,6 +33,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class SuperSmartAgent:
     def __init__(self):
+        self.model = YOLO("yolov8n.pt")
         self.graph = self._build_graph()
 
     def _build_graph(self):
@@ -129,7 +130,6 @@ class SuperSmartAgent:
             return state
 
         ###################
-        model = YOLO("yolov8n.pt")  # lightweight object detector
         
         def contains_youtube_url(text):
             return re.search(r"(https?://)?(www\.)?(youtube\.com|youtu\.be)/", text) is not None
@@ -166,7 +166,7 @@ class SuperSmartAgent:
 
         
         def use_video_analysis(state):        
-            url_match = re.search(r"(https?://[^\s]+)", state["question"])
+            url_match = re.search(r"(https?://(?:www\.)?(?:youtube\.com/watch\?v=|youtu\.be/)[\w\-]+)", state["question"])
             url = url_match.group(1).strip() if url_match else None
         
             if not url:
@@ -181,8 +181,9 @@ class SuperSmartAgent:
                 if not stream:
                     state["response"] = "No downloadable video stream found."
                     return state
-                temp_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
-                stream.download(filename=temp_file.name)
+                with tempfile.TemporaryDirectory() as tmpdirname:
+                    temp_path = os.path.join(tmpdirname, "video.mp4")
+                    stream.download(filename=temp_path)
         
                 cap = cv2.VideoCapture(temp_file.name)
                 frames = []
@@ -199,7 +200,7 @@ class SuperSmartAgent:
         
                 object_counts = []
                 for frame in frames:
-                    results = model(frame)
+                    results = self.model(frame)
                     count = len(results[0].boxes)
                     object_counts.append(count)