Update huggingface_exact_approach.py

huggingface_exact_approach.py

@@ -66,7 +66,7 @@ class VideoHighlightDetector:
     
     def analyze_video_content(self, video_path: str) -> str:
         """Analyze video content to determine its type and description."""
-        system_message = "You are a helpful assistant that can understand videos. Describe what type of video this is and what's happening in it."
+        system_message = "You are a helpful assistant that can understand videos. Give a concise, accurate 4-5 sentence description."
         messages = [
             {
                 "role": "system",
@@ -76,7 +76,7 @@ class VideoHighlightDetector:
                 "role": "user",
                 "content": [
                     {"type": "video", "path": video_path},
-                    {"type": "text", "text": "What type of video is this and what's happening in it? Be specific about the content type and general activities you observe."}
+                    {"type": "text", "text": "Describe this video in 4-5 sentences. Be accurate, concrete, and avoid guessing names/places unless clearly visible."}
                 ]
             }
         ]
@@ -95,12 +95,12 @@ class VideoHighlightDetector:
     def determine_highlights(self, video_description: str, prompt_num: int = 1) -> str:
         """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 other videos of this type.",
-            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. Highlights should be rare and important events in the video in question."
+            1: "You are a highlight editor. Return exactly 4-5 sentences describing the most important highlight moments to look for. Be specific, realistic, and avoid fabricated names or places.",
+            2: "You are a visual-language assistant for creating highlight reels. Return exactly 4-5 sentences describing rare, important highlight-worthy moments. Be concrete and avoid invented details."
         }
         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:"
+            1: "Based on this video description, describe the top highlight moments to look for in exactly 4-5 sentences.",
+            2: "Based on this video description, describe dramatic moments that would make compelling highlights in exactly 4-5 sentences."
         }
         
         
@@ -522,4 +522,4 @@ def main():
         traceback.print_exc()
 
 if __name__ == "__main__":
-    main()
+    main()