Update main.py

main.py

@@ -28,6 +28,24 @@ def get_file_size(url: str):
         return int(r.headers.get('Content-Length', 0))
     except: return 0
 
+# (Added from original) Required for the Force Encoding logic to work properly
+def download_temp_file(url: str, suffix: str):
+    """ဗီဒီယို သို့မဟုတ် အသံ stream ကို server ထဲသို့ ဒေါင်းလုဒ်ဆွဲခြင်း"""
+    if not url: return None
+    path = os.path.join(TEMP_DIR, f"{uuid.uuid4()}{suffix}")
+    try:
+        headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'}
+        r = requests.get(url, stream=True, timeout=30, headers=headers)
+        if r.status_code == 200:
+            with open(path, 'wb') as f:
+                for chunk in r.iter_content(chunk_size=1024*1024):
+                    f.write(chunk)
+            if os.path.getsize(path) > 1024:
+                return path
+    except: pass
+    if os.path.exists(path): os.remove(path)
+    return None
+
 @app.post("/download-private")
 async def download_private_fb(payload: dict = Body(...)):
     html_source = payload.get("html", "")
@@ -62,20 +80,48 @@ async def download_private_fb(payload: dict = Body(...)):
     }
 
 @app.get("/process-video")
-async def process_video(url: str, res: str, background_tasks: BackgroundTasks):
+async def process_video(url: str, res: str, background_tasks: BackgroundTasks, audio_url: str = None):
     # အသုံးပြုသူက Convert လုပ်ခိုင်းသည့်အခါမှသာ Processing လုပ်ပေးမည့် logic
     file_id = str(uuid.uuid4())
     out_file = os.path.join(TEMP_DIR, f"{file_id}.mp4")
     
     # ဤနေရာတွင် အရှေ့က ကျွန်တော်တို့ အောင်မြင်ခဲ့သော Force Encoding Logic ကို သုံးပါမယ်
-    # (မှတ်ချက် - ကုဒ်ရှည်မည်စိုး၍ အနှစ်ချုပ်ထားပါသည်)
-    target_h = res.replace("p", "")
-    subprocess.run([
-        'ffmpeg', '-y', '-i', url, 
-        '-vf', f'scale=w="trunc(oh*a/2)*2":h={target_h}',
-        '-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '24',
-        '-c:a', 'aac', '-b:a', '128k', '-movflags', '+faststart', out_file
-    ])
+    target_h = res.replace("p", "") if res else "720"
+    v_tmp = download_temp_file(url, ".mp4")
+    a_tmp = download_temp_file(audio_url, ".mp3") if audio_url else None
+
+    try:
+        if v_tmp:
+            # Merging Streams & Forcing H.264 (avc1) + AAC...
+            if a_tmp:
+                cmd = [
+                    'ffmpeg', '-y', '-i', v_tmp, '-i', a_tmp,
+                    '-map', '0:v:0', '-map', '1:a:0',
+                    '-vf', f'scale=w="trunc(oh*a/2)*2":h={target_h}',
+                    '-c:v', 'libx264', '-preset', 'fast', '-crf', '23', 
+                    '-c:a', 'aac', '-b:a', '128k',
+                    '-movflags', '+faststart', out_file
+                ]
+            # Video only mode: Converting to H.264...
+            else:
+                cmd = [
+                    'ffmpeg', '-y', '-i', v_tmp, 
+                    '-vf', f'scale=w="trunc(oh*a/2)*2":h={target_h}',
+                    '-c:v', 'libx264', '-preset', 'fast', '-crf', '23', 
+                    '-c:a', 'aac', '-movflags', '+faststart', out_file
+                ]
+            
+            subprocess.run(cmd, check=True, capture_output=True)
+
+            if v_tmp: os.remove(v_tmp)
+            if a_tmp: os.remove(a_tmp)
+        else:
+            raise Exception("Video stream download failed.")
+    except Exception as e:
+        # Error တက်ပါက Temp File အပျက်အစီးများကို ရှင်းလင်းမည်
+        if v_tmp and os.path.exists(v_tmp): os.remove(v_tmp)
+        if a_tmp and os.path.exists(a_tmp): os.remove(a_tmp)
+        raise HTTPException(status_code=500, detail=f"Processing error: {str(e)}")
     
     space_id = os.getenv('SPACE_ID', '').replace("/", "-")
     return {"download_url": f"https://{space_id}.hf.space/get_file?id={file_id}.mp4"}