Update app.py

app.py

@@ -164,80 +164,54 @@ from dotenv import load_dotenv
 load_dotenv()
 
 def create_client():
-    """Create a client connection to the private space"""
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
         raise ValueError("HF_TOKEN environment variable not set")
+    return Client("wxcyn/video-analysis", hf_token=hf_token)
 
-    return Client(
-        "wxcyn/video-analysis",
-        hf_token=hf_token
+def process_video(video_file, question, password, progress=gr.Progress()):
+    if video_file is None:
+        raise gr.Error("No video file uploaded.")
+
+    local_file_path = video_file
+    if not os.path.exists(local_file_path):
+        raise gr.Error("Uploaded file could not be accessed.")
+
+    progress(0.2, desc="Connecting to video analysis service...")
+    client = create_client()
+
+    progress(0.4, desc="Uploading and processing video...")
+    wrapped_file = handle_file(local_file_path)  # Convert to FileData
+
+    result = client.predict(
+        video_file=wrapped_file,
+        question=question,
+        password=password,
+        api_name="/process_video"
     )
 
-def process_video(video_file, question, password, progress=gr.Progress()):
-    """Process video by calling the private space API"""
-    try:
-        if video_file is None:
-            raise gr.Error("No video file uploaded.")
-
-        local_file_path = video_file
-        if not local_file_path or not os.path.exists(local_file_path):
-            raise gr.Error("Uploaded file could not be accessed.")
-
-        # Show initial progress
-        progress(0.2, desc="Connecting to video analysis service...")
-        client = create_client()
-
-        # Convert the local file path to a FileData object
-        progress(0.4, desc="Uploading and processing video...")
-        wrapped_file = handle_file(local_file_path)
-
-        # Call the private API with the properly formatted file input
-        result = client.predict(
-            video_file=wrapped_file,
-            question=question,
-            password=password,
-            api_name="/process_video"
-        )
+    progress(0.8, desc="Processing results...")
 
-        progress(0.8, desc="Processing results...")
-
-        # Expecting result in the form (analysis_text, video_clip_info)
-        if isinstance(result, tuple) and len(result) == 2:
-            analysis_text, video_clip_info = result
-
-            # Check if video_clip_info is a dictionary or a string path
-            if isinstance(video_clip_info, dict):
-                # The dict should have a 'name' key pointing to the local path of the returned clip
-                video_path = video_clip_info.get("name")
-                if video_path and os.path.exists(video_path):
-                    return analysis_text, video_path
-                else:
-                    # If no valid video clip path, just return the text
-                    return analysis_text, None
-            else:
-                # If it's a string and that path exists, return it directly
-                if isinstance(video_clip_info, str) and os.path.exists(video_clip_info):
-                    return analysis_text, video_clip_info
-
-                # Otherwise, no video clip
-                return analysis_text, None
-        else:
-            # If result doesn't match the expected structure, just return what we got
-            return str(result), None
-
-    except gr.Error as e:
-        # Known Gradio errors are just re-raised
-        raise e
-    except Exception as e:
-        # Unexpected errors are wrapped in a gr.Error
-        raise gr.Error(f"Error processing video: {str(e)}")
+    if isinstance(result, tuple) and len(result) == 2:
+        analysis_text, video_clip_info = result
+
+        # If video_clip_info is a dict with 'name', it's a FileData dict
+        if isinstance(video_clip_info, dict) and "name" in video_clip_info:
+            video_path = video_clip_info["name"]
+            if os.path.exists(video_path):
+                return analysis_text, video_path
+        # If video_clip_info is a string path, check and return it directly
+        elif isinstance(video_clip_info, str) and os.path.exists(video_clip_info):
+            return analysis_text, video_clip_info
+
+        # No valid video clip found
+        return analysis_text, None
+    else:
+        return str(result), None
 
 def create_interface():
-    """Create the Gradio interface"""
     with gr.Blocks(title="Public Video Analysis Interface") as interface:
-        gr.Markdown("# Video Content Analysis (1 file under 2.5 minutes)")
-
+        gr.Markdown("# Public Video Content Analysis")
         with gr.Row():
             with gr.Column(scale=2):
                 password = gr.Textbox(
@@ -246,7 +220,6 @@ def create_interface():
                     info="Enter your access password",
                     interactive=True
                 )
-                # Use "filepath" so that `video_file` is a local path
                 video_file = gr.File(
                     label="Upload Video File (up to 30MB)",
                     file_types=['video'],
@@ -266,10 +239,7 @@ def create_interface():
                     lines=15,
                     interactive=False
                 )
-                video_output = gr.Video(
-                    label="Relevant Video Clip",
-                    interactive=False
-                )
+                video_output = gr.Video(label="Relevant Video Clip", interactive=False)
 
         submit_btn.click(
             fn=process_video,
@@ -280,16 +250,13 @@ def create_interface():
     return interface
 
 if __name__ == "__main__":
-    # Create required directories
     Path('data').mkdir(exist_ok=True)
     Path('temp').mkdir(exist_ok=True)
-
-    # Create and launch interface
     interface = create_interface()
-    interface.queue()  # Enable queuing for concurrency
+    interface.queue()
     interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True,  # Enable verbose error reporting
-        max_threads=140    # Increased thread limit
+        show_error=True,
+        max_threads=140
     )