two highlights path

app.py

@@ -87,8 +87,13 @@ class VideoHighlightDetector:
         """Determine what constitutes highlights based on video description with different prompts."""
         system_prompts = {
             1: "You are a highlight editor. List archetypal dramatic moments that would make compelling highlights if they appear in the video. Each moment should be specific enough to be recognizable but generic enough to potentially exist in any video of this type.",
-            2: "You are a highlight editor focusing on subtle and nuanced moments. List quieter, character-driven, or atmospheric moments that would make interesting highlights. Focus on moments that might be overlooked by traditional highlight detection but add depth to the story."
+            2: "You are a helpful visual-language assistant that can understand videos and edit. You are tasked helping the user to create highlight reels for videos. Generally, highlights should be relatively rare and important events in the video in question."
         }
+        user_prompts = {
+            1: "List potential highlight moments to look for in this video:",
+            2: "List dramatic moments that would make compelling highlights if they appear in the video. Each moment should be specific enough to be recognizable but generic enough to potentially exist in any video of this type:"
+        }
+
         
         messages = [
             {
@@ -97,7 +102,7 @@ class VideoHighlightDetector:
             },
             {
                 "role": "user",
-                "content": [{"type": "text", "text": f"""Here is a description of a video:\n\n{video_description}\n\nList potential highlight moments to look for in this video:"""}]
+                "content": [{"type": "text", "text": f"""Here is a description of a video:\n\n{video_description}\n\n{user_prompts[prompt_num]}"""}]
             }
         ]
 
@@ -290,7 +295,7 @@ def create_ui(examples_path: str, model_path: str):
 
                 detector = VideoHighlightDetector(
                     model_path=model_path,
-                    batch_size=8
+                    batch_size=16
                 )
 
                 yield [
@@ -329,7 +334,6 @@ def create_ui(examples_path: str, model_path: str):
                 total_segments = int(duration / segment_length)
 
                 for start_time in range(0, int(duration), int(segment_length)):
-                    segments_processed += 1
                     progress = int((segments_processed / total_segments) * 100)
                     
                     yield [
@@ -368,6 +372,8 @@ def create_ui(examples_path: str, model_path: str):
                     
                     # Clean up segment file
                     os.remove(segment_path)
+                    segments_processed += 1
+
 
                 # Remove temp directory
                 os.rmdir(temp_dir)