Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import os
 import subprocess
-import ffmpeg
+import ffmpeg  # Import the ffmpeg-python library
 
 # Directories for uploads and processing
 UPLOAD_FOLDER = 'uploads'
@@ -9,61 +9,58 @@ PROCESSED_FOLDER = 'processed'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(PROCESSED_FOLDER, exist_ok=True)
 
-def get_new_filename(original_filename):
-    """Generate a new filename with an incremented number."""
-    base, ext = os.path.splitext(original_filename)
-    i = 1
-    new_filename = f"{base}_{i}{ext}"
-    while os.path.exists(os.path.join(UPLOAD_FOLDER, new_filename)):
-        i += 1
-        new_filename = f"{base}_{i}{ext}"
-    return new_filename
+def save_uploaded_file(uploaded_file):
+    """Save uploaded file to a designated folder without renaming it unless a file with the same name exists."""
+    file_path = os.path.join(UPLOAD_FOLDER, uploaded_file.name)
+    # If a file with the same name already exists, generate a unique name
+    if os.path.exists(file_path):
+        base, ext = os.path.splitext(uploaded_file.name)
+        i = 1
+        while os.path.exists(os.path.join(UPLOAD_FOLDER, f"{base}_{i}{ext}")):
+            i += 1
+        file_path = os.path.join(UPLOAD_FOLDER, f"{base}_{i}{ext}")
+    with open(file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+    return file_path
+
+def run_ffmpeg(command):
+    """Run the FFmpeg command and capture the output."""
+    try:
+        result = subprocess.run(command, shell=True, capture_output=True, text=True)
+        return result.stdout, result.stderr
+    except Exception as e:
+        return "", str(e)
 
 # Streamlit UI
-st.title('FFmpeg Command Executor with ffmpeg-python')
+st.title("FFmpeg Command Runner")
 
-# File upload
-uploaded_file = st.file_uploader("Upload a file", type=None)  # Accept any file type
-if uploaded_file:
-    new_filename = get_new_filename(uploaded_file.name)
-    file_path = os.path.join(UPLOAD_FOLDER, new_filename)
-    with open(file_path, 'wb') as f:
-        f.write(uploaded_file.getbuffer())
-    st.success(f'File uploaded successfully as {new_filename}!')
+# File uploader
+uploaded_file = st.file_uploader("Upload a file", type=None)
 
-# FFmpeg command input
-ffmpeg_command = st.text_area("Enter FFmpeg command", placeholder="e.g., -vf scale=640:480 output.mp4")
+# Text area for FFmpeg commands
+ffmpeg_command = st.text_area("Enter your FFmpeg command", "")
 
-if st.button('Run Command'):
-    if not uploaded_file:
-        st.error("Please upload a file before running the command.")
-    elif not ffmpeg_command:
-        st.error("Please enter an FFmpeg command.")
-    else:
-        # Build paths
-        input_file = os.path.join(UPLOAD_FOLDER, new_filename)
-        output_file = os.path.join(PROCESSED_FOLDER, 'output.mp4')
-        
-        # Prepare FFmpeg command
-        command = f"ffmpeg -i {input_file} {ffmpeg_command}"
-        
-        # Replace 'output' placeholder in the command with the actual output file path
-        command = command.replace('output', output_file)
+# Execute button
+if st.button("Run FFmpeg Command"):
+    if uploaded_file:
+        # Save the uploaded file
+        uploaded_file_path = save_uploaded_file(uploaded_file)
+        st.write(f"File saved at: {uploaded_file_path}")
+
+        # Replace placeholder with the uploaded file path in the FFmpeg command
+        command_to_run = ffmpeg_command.replace("<input_file>", uploaded_file_path)
 
-        try:
-            # Execute FFmpeg command
-            result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
-            st.success("Command executed successfully!")
-            
-            # Display logs
-            st.subheader('FFmpeg Logs:')
-            st.code(result.stdout)
-            
-            # Display video player
-            if os.path.exists(output_file):
-                st.subheader('Processed Video:')
-                st.video(output_file)
-            else:
-                st.warning("No video file found. Please check the command.")
-        except subprocess.CalledProcessError as e:
-            st.error(f"Error executing command: {e}")
+        # Run the command
+        stdout, stderr = run_ffmpeg(command_to_run)
+
+        # Display the logs
+        st.text_area("FFmpeg Output", stdout)
+        st.text_area("FFmpeg Errors", stderr)
+
+        # Check for a successful FFmpeg process and show the video player if the output file exists
+        output_path = os.path.join(PROCESSED_FOLDER, "output.mp4")  # Change as needed
+        if os.path.exists(output_path):
+            st.video(output_path)
+
+    else:
+        st.write("Please upload a file before running the command.")