Update modules/app.py

modules/app.py

@@ -84,6 +84,30 @@ async def add_audio_to_image(request: Request):
         print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
+@app.post("/add_audio_to_video")
+async def add_audio_to_video(request: Request):
+    try:
+        # Generate a unique filename
+        output_filename = f"{uuid.uuid4()}.mp4"
+        output_path = os.path.join(OUTPUT_DIR, output_filename)
+
+        # Call the modal API with the request data and download the output file
+        data = await request.json()
+        async with aiohttp.ClientSession() as session:
+            async with session.post(f"{MODAL_BASE_URL}/add_audio_to_video", json=data) as response:
+                if response.status != 200:
+                    raise HTTPException(status_code=500, detail="Failed to process request")
+                output_data = await response.read()
+                async with aiofiles.open(output_path, "wb") as f:
+                    await f.write(output_data)
+
+        # Return the URL path to the output file
+        return f"https://sxqib-api.hf.space/output/{output_filename}"
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+        print(traceback.format_exc())
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
+
 @app.post("/concatenate_videos")
 async def concatenate_videos(request: Request):
     try: