Update app.py

app.py

@@ -58,14 +58,12 @@ allowed_medias = [
     ".3gpp",
 ]
 
-
 def get_files_infos(files):
     results = []
     for file in files:
         file_path = Path(file.name)
         info = {}
         info["size"] = os.path.getsize(file_path)
-        # Sanitize filename by replacing spaces with underscores
         info["name"] = file_path.name.replace(" ", "_")
         file_extension = file_path.suffix
 
@@ -84,39 +82,21 @@ def get_files_infos(files):
             info["duration"] = audio.duration
             info["audio_channels"] = audio.nchannels
             audio.close()
-        elif file_extension in (
-            ".png",
-            ".jpg",
-            ".jpeg",
-            ".tiff",
-            ".bmp",
-            ".gif",
-            ".svg",
-        ):
+        elif file_extension in (".png", ".jpg", ".jpeg", ".tiff", ".bmp", ".gif", ".svg"):
             info["type"] = "image"
             img = Image.open(file.name)
             info["dimensions"] = "{}x{}".format(img.size[0], img.size[1])
         results.append(info)
     return results
 
-
 def get_completion(prompt, files_info, top_p, temperature, model_choice):
-    # Create table header
     files_info_string = "| Type | Name | Dimensions | Duration | Audio Channels |\n"
     files_info_string += "|------|------|------------|-----------|--------|\n"
 
-    # Add each file as a table row
     for file_info in files_info:
         dimensions = file_info.get("dimensions", "-")
-        duration = (
-            f"{file_info.get('duration', '-')}s" if "duration" in file_info else "-"
-        )
-        audio = (
-            f"{file_info.get('audio_channels', '-')} channels"
-            if "audio_channels" in file_info
-            else "-"
-        )
-
+        duration = f"{file_info.get('duration', '-')}s" if "duration" in file_info else "-"
+        audio = f"{file_info.get('audio_channels', '-')} channels" if "audio_channels" in file_info else "-"
         files_info_string += f"| {file_info['type']} | {file_info['name']} | {dimensions} | {duration} | {audio} |\n"
 
     messages = [
@@ -161,7 +141,6 @@ YOUR FFMPEG COMMAND:
         },
     ]
     try:
-        # Print the complete prompt
         print("\n=== COMPLETE PROMPT ===")
         for msg in messages:
             print(f"\n[{msg['role'].upper()}]:")
@@ -184,11 +163,8 @@ YOUR FFMPEG COMMAND:
             max_tokens=2048,
         )
         content = completion.choices[0].message.content
-        # Extract command from code block if present
         if "```" in content:
-            # Find content between ```sh or ```bash and the next ```
             import re
-
             command = re.search(r"```(?:sh|bash)?\n(.*?)\n```", content, re.DOTALL)
             if command:
                 command = command.group(1).strip()
@@ -197,32 +173,20 @@ YOUR FFMPEG COMMAND:
         else:
             command = content.replace("\n", "")
 
-        # remove output.mp4 with the actual output file path
         command = command.replace("output.mp4", "")
-
         return command
     except Exception as e:
         raise Exception("API Error")
 
-
-def update(
-    files,
-    prompt,
-    top_p=1,
-    temperature=1,
-    model_choice="Qwen/Qwen2.5-Coder-32B-Instruct",
-):
+def update(files, prompt, top_p=1, temperature=1, model_choice="Qwen/Qwen2.5-Coder-32B-Instruct"):
     if prompt == "":
         raise gr.Error("Please enter a prompt.")
 
     files_info = get_files_infos(files)
-    # disable this if you're running the app locally or on your own server
     for file_info in files_info:
         if file_info["type"] == "video":
             if file_info["duration"] > 120:
-                raise gr.Error(
-                    "Please make sure all videos are less than 2 minute long."
-                )
+                raise gr.Error("Please make sure all videos are less than 2 minute long.")
         if file_info["size"] > 100000000:
             raise gr.Error("Please make sure all files are less than 100MB in size.")
 
@@ -230,25 +194,18 @@ def update(
     while attempts < 2:
         print("ATTEMPT", attempts)
         try:
-            command_string = get_completion(
-                prompt, files_info, top_p, temperature, model_choice
-            )
-            print(
-                f"""///PROMTP {prompt} \n\n/// START OF COMMAND ///:\n\n{command_string}\n\n/// END OF COMMAND ///\n\n"""
-            )
+            command_string = get_completion(prompt, files_info, top_p, temperature, model_choice)
+            print(f"""///PROMTP {prompt} \n\n/// START OF COMMAND ///:\n\n{command_string}\n\n/// END OF COMMAND ///\n\n""")
 
-            # split command string into list of arguments
             args = shlex.split(command_string)
             if args[0] != "ffmpeg":
                 raise Exception("Command does not start with ffmpeg")
             temp_dir = tempfile.mkdtemp()
-            # copy files to temp dir with sanitized names
             for file in files:
                 file_path = Path(file.name)
                 sanitized_name = file_path.name.replace(" ", "_")
                 shutil.copy(file_path, Path(temp_dir) / sanitized_name)
 
-            # test if ffmpeg command is valid dry run
             ffmpg_dry_run = subprocess.run(
                 args + ["-f", "null", "-"],
                 stderr=subprocess.PIPE,
@@ -260,16 +217,12 @@ def update(
             else:
                 print("Command is not valid. Error output:")
                 print(ffmpg_dry_run.stderr)
-                raise Exception(
-                    "FFMPEG generated command is not valid. Please try something else."
-                )
+                raise Exception("FFMPEG generated command is not valid. Please try something else.")
 
             output_file_name = f"output_{uuid.uuid4()}.mp4"
             output_file_path = str((Path(temp_dir) / output_file_name).resolve())
             final_command = args + ["-y", output_file_path]
-            print(
-                f"\n=== EXECUTING FFMPEG COMMAND ===\nffmpeg {' '.join(final_command[1:])}\n"
-            )
+            print(f"\n=== EXECUTING FFMPEG COMMAND ===\nffmpeg {' '.join(final_command[1:])}\n")
             subprocess.run(final_command, cwd=temp_dir)
             generated_command = f"### Generated Command\n```bash\nffmpeg {' '.join(args[1:])} -y output.mp4\n```"
             return output_file_path, gr.update(value=generated_command)
@@ -279,22 +232,28 @@ def update(
                 print("FROM UPDATE", e)
                 raise gr.Error(e)
 
-
-# Custom login system
 def check_password(password):
     return password == PASSWORD
 
+# Custom CSS for mobile responsiveness
+custom_css = """
+@media (max-width: 768px) {
+    .gradio-container {
+        flex-direction: column;
+    }
+    .gradio-column {
+        width: 100% !important;
+    }
+}
+"""
 
-# Gradio app with password protection
-with gr.Blocks() as demo:
-    # Login interface (visible by default)
+with gr.Blocks(css=custom_css) as demo:
     with gr.Row(visible=True) as login_interface:
         with gr.Column():
             password_input = gr.Textbox(label="Enter Password", type="password")
             login_button = gr.Button("Login")
             login_status = gr.Textbox(label="Login Status", interactive=False)
 
-    # Main app interface (hidden by default)
     with gr.Row(visible=False) as main_interface:
         gr.Markdown(
             """
@@ -303,45 +262,43 @@ with gr.Blocks() as demo:
         """,
             elem_id="header",
         )
-        with gr.Row():
-            with gr.Column():
-                user_files = gr.File(
-                    file_count="multiple",
-                    label="Media files",
-                    file_types=allowed_medias,
+        with gr.Column():
+            user_files = gr.File(
+                file_count="multiple",
+                label="Media files",
+                file_types=allowed_medias,
+            )
+            user_prompt = gr.Textbox(
+                placeholder="eg: Remove the 3 first seconds of the video",
+                label="Instructions",
+            )
+            btn = gr.Button("Run")
+            with gr.Accordion("Parameters", open=False):
+                model_choice = gr.Radio(
+                    choices=list(MODELS.keys()),
+                    value=list(MODELS.keys())[0],
+                    label="Model",
                 )
-                user_prompt = gr.Textbox(
-                    placeholder="eg: Remove the 3 first seconds of the video",
-                    label="Instructions",
+                top_p = gr.Slider(
+                    minimum=-0,
+                    maximum=1.0,
+                    value=0.7,
+                    step=0.05,
+                    interactive=True,
+                    label="Top-p (nucleus sampling)",
                 )
-                btn = gr.Button("Run")
-                with gr.Accordion("Parameters", open=False):
-                    model_choice = gr.Radio(
-                        choices=list(MODELS.keys()),
-                        value=list(MODELS.keys())[0],
-                        label="Model",
-                    )
-                    top_p = gr.Slider(
-                        minimum=-0,
-                        maximum=1.0,
-                        value=0.7,
-                        step=0.05,
-                        interactive=True,
-                        label="Top-p (nucleus sampling)",
-                    )
-                    temperature = gr.Slider(
-                        minimum=-0,
-                        maximum=5.0,
-                        value=0.1,
-                        step=0.1,
-                        interactive=True,
-                        label="Temperature",
-                    )
-            with gr.Column():
-                generated_video = gr.Video(
-                    interactive=False, label="Generated Video", include_audio=True
+                temperature = gr.Slider(
+                    minimum=-0,
+                    maximum=5.0,
+                    value=0.1,
+                    step=0.1,
+                    interactive=True,
+                    label="Temperature",
                 )
-                generated_command = gr.Markdown()
+            generated_video = gr.Video(
+                interactive=False, label="Generated Video", include_audio=True
+            )
+            generated_command = gr.Markdown()
 
             btn.click(
                 fn=update,
@@ -396,9 +353,15 @@ with gr.Blocks() as demo:
                 cache_examples=False,
             )
 
-        
+        with gr.Row():
+            gr.Markdown(
+                """
+                If you have idea to improve this please open a PR:
+
+                [![Open a Pull Request](https://huggingface.co/datasets/huggingface/badges/raw/main/open-a-pr-lg-light.svg)](https://huggingface.co/spaces/huggingface-projects/video-composer-gpt4/discussions)
+                """,
+            )
 
-    # Show/hide main interface based on login status
     def toggle_interface(password):
         if check_password(password):
             return gr.update(visible=False), gr.update(visible=True), "Login successful!"
@@ -411,6 +374,5 @@ with gr.Blocks() as demo:
         outputs=[login_interface, main_interface, login_status],
     )
 
-# Launch the app with embedding enabled
 demo.queue(default_concurrency_limit=200)
 demo.launch(show_api=False, ssr_mode=False, share=True, inline=True)