fixes

app.py

@@ -157,71 +157,73 @@ class VideoHighlightDetector:
         outputs = self.model.generate(**inputs, max_new_tokens=64, do_sample=False)
         response = self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("assistant: ")[1]
         return "yes" in response
-
 def create_xspf_playlist(video_path: str, segments: list, descriptions: list) -> str:
     """Create XSPF playlist from segments with descriptions."""
-    # Define namespaces
-    XSPF_NS = "http://xspf.org/ns/0/"
-    VLC_NS = "http://www.videolan.org/vlc/playlist/ns/0/"
+    # Register namespaces to avoid namespace prefix issues
+    ET.register_namespace('', "http://xspf.org/ns/0/")
+    ET.register_namespace('vlc', "http://www.videolan.org/vlc/playlist/ns/0/")
     
     # Create the root element with proper namespace
-    root = ET.Element("playlist", {
-        "xmlns": XSPF_NS,
-        "xmlns:vlc": VLC_NS,
+    root = ET.Element("{http://xspf.org/ns/0/}playlist", {
         "version": "1"
     })
     
     # Get video filename for the title
     video_filename = os.path.basename(video_path)
-    title = ET.SubElement(root, "title")
+    title = ET.SubElement(root, "{http://xspf.org/ns/0/}title")
     title.text = f"{video_filename} - Highlights"
     
-    tracklist = ET.SubElement(root, "trackList")
+    tracklist = ET.SubElement(root, "{http://xspf.org/ns/0/}trackList")
     
     for idx, ((start_time, end_time), description) in enumerate(zip(segments, descriptions)):
-        track = ET.SubElement(tracklist, "track")
+        track = ET.SubElement(tracklist, "{http://xspf.org/ns/0/}track")
         
-        location = ET.SubElement(track, "location")
+        location = ET.SubElement(track, "{http://xspf.org/ns/0/}location")
         location.text = f"file:///{video_filename}"
         
-        title = ET.SubElement(track, "title")
+        title = ET.SubElement(track, "{http://xspf.org/ns/0/}title")
         title.text = f"Highlight {idx + 1}: {description}"
         
-        annotation = ET.SubElement(track, "annotation")
+        annotation = ET.SubElement(track, "{http://xspf.org/ns/0/}annotation")
         annotation.text = description
         
-        start_meta = ET.SubElement(track, "meta")
+        start_meta = ET.SubElement(track, "{http://xspf.org/ns/0/}meta")
         start_meta.set("rel", "start")
         start_meta.text = format_duration(start_time)
         
-        end_meta = ET.SubElement(track, "meta")
+        end_meta = ET.SubElement(track, "{http://xspf.org/ns/0/}meta")
         end_meta.set("rel", "end")
         end_meta.text = format_duration(end_time)
     
-    # Add VLC extension
-    extension = ET.SubElement(root, "extension")
+    # Add VLC extension with proper namespace
+    extension = ET.SubElement(root, "{http://xspf.org/ns/0/}extension")
     extension.set("application", "http://www.videolan.org/vlc/playlist/0")
     
     for i in range(len(segments)):
-        item = ET.SubElement(extension, "{%s}item" % VLC_NS)
+        item = ET.SubElement(extension, "{http://www.videolan.org/vlc/playlist/ns/0/}item")
         item.set("tid", str(i))
     
     # Convert to string with pretty printing
     xml_str = minidom.parseString(ET.tostring(root, encoding='unicode')).toprettyxml(indent="    ")
     
-    # Clean up any potential namespace declaration issues
-    xml_str = xml_str.replace('xmlns:ns0="http://www.videolan.org/vlc/playlist/ns/0/"', '')
-    xml_str = xml_str.replace('ns0:', 'vlc:')
-    
     return xml_str
 
 def create_ui(examples_path: str, model_path: str):
     examples_data = load_examples(examples_path)
 
     with gr.Blocks() as app:
-        gr.Markdown("# Video Highlight Playlist Generator")
-        gr.Markdown("Upload a video and get an XSPF playlist of highlights!")
+        # gr.Markdown("# VLC Highlight Generator")
+        # gr.Markdown("Upload a video and get a list of highlights!")
+        with gr.Row():
+            gr.HTML("""
+                <div style="display: flex; align-items: center; gap: 10px;">
+                    <img src="https://upload.wikimedia.org/wikipedia/commons/3/38/VLC_icon.png" style="width: 40px; height: 40px;"/>
+                    <h1 style="margin: 0;">VLC Highlight Generator</h1>
+                </div>
+            """)
         
+        gr.Markdown("Upload a video and get a list of highlights!")
+         
         with gr.Row():
             with gr.Column(scale=1):
                 input_video = gr.Video(