Create main.py

main.py

@@ -0,0 +1,118 @@
+# main.py
+import gradio as gr
+from fastapi import FastAPI, File, UploadFile, Form
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+import os
+import tempfile
+import shutil
+import traceback
+import logging
+from merge_utils import merge_videos_and_audios  # move your merge logic to merge_utils.py for clean code!
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# --- FASTAPI SETUP ---
+fastapi_app = FastAPI(title="Video & Audio Merger API")
+
+# CORS for n8n to call the API
+fastapi_app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # You can restrict to your domain if needed
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# FastAPI endpoint to accept binary files from n8n
+@fastapi_app.post("/merge")
+async def merge_endpoint(
+    files: list[UploadFile] = File(...),
+    orig_vol: float = Form(1.0),
+    music_vol: float = Form(0.5),
+):
+    try:
+        logger.info(f"Received {len(files)} files from n8n")
+
+        # Save files to temp dir
+        temp_dir = tempfile.mkdtemp()
+        saved_files = []
+
+        for upload_file in files:
+            filename = os.path.basename(upload_file.filename)
+            file_path = os.path.join(temp_dir, filename)
+            with open(file_path, "wb") as f:
+                content = await upload_file.read()
+                f.write(content)
+            saved_files.append(file_path)
+            logger.info(f"Saved file: {file_path}")
+
+        # Split files into video and audio
+        video_files = [f for f in saved_files if f.lower().endswith(".mp4")]
+        audio_files = [f for f in saved_files if f.lower().endswith((".mp3", ".wav"))]
+
+        # Run merge
+        output_path = merge_videos_and_audios(
+            video_files=video_files,
+            audio_files=audio_files,
+            orig_vol=orig_vol,
+            music_vol=music_vol,
+            temp_dir=temp_dir,
+        )
+
+        # Return result as path or error
+        if isinstance(output_path, str) and output_path.startswith("Error"):
+            return {"success": False, "error": output_path}
+
+        # Move output to persistent location (optional)
+        public_output_dir = "./outputs"
+        os.makedirs(public_output_dir, exist_ok=True)
+        final_output_path = os.path.join(public_output_dir, os.path.basename(output_path))
+        shutil.move(output_path, final_output_path)
+        logger.info(f"Final output moved to {final_output_path}")
+
+        return {
+            "success": True,
+            "output_url": f"/outputs/{os.path.basename(final_output_path)}"
+        }
+
+    except Exception as e:
+        error_msg = f"Exception: {str(e)}\n{traceback.format_exc()}"
+        logger.error(error_msg)
+        return {"success": False, "error": error_msg}
+
+# --- GRADIO SETUP ---
+with gr.Blocks(title="Video and Audio Merger UI") as gradio_app:
+    gr.Markdown("## Video and Audio Merger")
+    file_input = gr.File(label="Upload Files", type="binary", file_count="multiple")
+    orig_vol_input = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, step=0.05, label="Original Video Audio Volume")
+    music_vol_input = gr.Slider(minimum=0.0, maximum=1.0, value=0.5, step=0.05, label="Background Audio Volume")
+    output_video = gr.Video(label="Merged Video")
+    output_audio = gr.Audio(label="Merged Audio")
+    merge_button = gr.Button("Merge Files")
+
+    def gradio_merge_files(file_binaries, orig_vol, music_vol):
+        # You can call your existing function here (same as before)
+        pass  # Keep your old gradio_merge_files() here!
+
+    merge_button.click(
+        fn=gradio_merge_files,
+        inputs=[file_input, orig_vol_input, music_vol_input],
+        outputs=[output_video, output_audio],
+    )
+
+# --- COMBINE BOTH ---
+if __name__ == "__main__":
+    import threading
+
+    # Run FastAPI server in background
+    def run_fastapi():
+        uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)
+
+    threading.Thread(target=run_fastapi).start()
+
+    # Run Gradio app on another port (HuggingFace will pick this)
+    gradio_app.queue(api_open=True)
+    gradio_app.launch(server_port=7860, share=True)