Update main.py

main.py

@@ -1,5 +1,3 @@
-import yt_dlp
-from fastapi import FastAPI, HTTPException, Query, BackgroundTasks, Body
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import FileResponse
 from starlette.background import BackgroundTask
@@ -41,117 +39,128 @@ def clean_fb_url(url: str):
     except: pass
     return urllib.parse.unquote(url).replace('&', '&')
 
+def get_ffprobe_info(file_path):
+
+
+    try:
+        cmd = ['ffprobe', '-v', 'error', '-show_entries', 'stream=codec_type', '-of', 'json', file_path]
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        data = json.loads(result.stdout)
+        return [s['codec_type'] for s in data.get('streams', [])]
+    except: return []
+
 @app.get("/")
 def root():
-    return {"message": "Bumiz FB Pro: Real HQ Engine Active"}
+    return {"message": "Bumiz Facebook Pro API Active"}
+
 
-# --- ၁။ ANALYZE: QUALITY အစုံရှာခြင်း (Duplicate Filter ပါဝင်သည်) ---
-@app.post("/analyze-private")
-async def analyze_private_fb(payload: dict = Body(...)):
+@app.post("/download-private")
+async def download_private_fb(payload: dict = Body(...)):
     html_source = payload.get("html", "")
-    if not html_source:
-        return {"success": False, "error": "Source is empty"}
+    target_quality = payload.get("quality", "original") # original, 720, 480, 240
+
 
-    # Video Representations ရှာဖွေခြင်း
-    video_matches = re.findall(r'"height":(\d+),.*?"base_url":"([^"]+)"', html_source)
     
-    unique_qualities = {}
-    for height, url in video_matches:
-        h = int(height)
-        u = clean_fb_url(url)
-        if u and "fbcdn.net" in u:
-            # တူညီတဲ့ height တွေကို တစ်ခုပဲ ယူမယ် (အကြည်ဆုံး link ကိုပဲ သိမ်းမယ်)
-            if h not in unique_qualities:
-                unique_qualities[h] = u
-
-    # Result ပြန်ပို့ရန် format ပြင်ခြင်း
-    final_qualities = []
-    for h in sorted(unique_qualities.keys(), reverse=True):
-        final_qualities.append({
-            "label": f"{h}p Quality",
-            "url": unique_qualities[h],
-            "height": h,
-            "type": "hd" if h >= 720 else "sd"
-        })
-
-    # Audio Link ရှာဖွေခြင်း (အသံသီးသန့် အကွက်ထဲမှာ အရင်ရှာမယ်)
+    if not html_source:
+        return {"success": False, "error": "HTML Source is empty."}
+
+
+
+
+
+    # Video & Audio Extraction Logic
+    video_url = None
     audio_url = None
-    # Method A: DASH Manifest Audio
-    a_match = re.search(r'"mime_type":"audio/[^"]+","base_url":"([^"]+)"', html_source)
-    if a_match:
-        audio_url = clean_fb_url(a_match.group(1))
-    else:
-        # Method B: Alternative Audio key
-        a_match_alt = re.search(r'"audio":\[\{"url":"([^"]+)"', html_source)
-        if a_match_alt:
-            audio_url = clean_fb_url(a_match_alt.group(1))
-
-    if not final_qualities:
-        return {"success": False, "error": "No valid video streams found."}
-
-    return {"success": True, "qualities": final_qualities, "audio_url": audio_url}
-
-# --- ၂။ PROCESS: အကြည်ဆုံး ရအောင် ပေါင်းစပ်ခြင်း ---
-@app.post("/process-private")
-async def process_private_fb(background_tasks: BackgroundTasks, payload: dict = Body(...)):
-    video_url = payload.get("video_url")
-    audio_url = payload.get("audio_url")
+
+    # HD သို့မဟုတ် Progressive Link ကို အရင်ရှာခြင်း
+    hd_match = re.search(r'"browser_native_hd_url":"([^"]+)"', html_source)
+    sd_match = re.search(r'"browser_native_sd_url":"([^"]+)"', html_source)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
     
-    file_id = str(uuid.uuid4())
+    if hd_match: video_url = clean_fb_url(hd_match.group(1))
+    elif sd_match: video_url = clean_fb_url(sd_match.group(1))
+
+    # Separate Streams ရှာဖွေခြင်း
+    v_matches = re.findall(r'"mime_type":"video[^"]+".*?"base_url":"([^"]+)"', html_source)
+    a_matches = re.findall(r'"mime_type":"audio[^"]+".*?"base_url":"([^"]+)"', html_source)
+
+    if not video_url and v_matches: video_url = clean_fb_url(v_matches[0])
+    if a_matches: audio_url = clean_fb_url(a_matches[0])
+
+    if not video_url:
+        return {"success": False, "error": "Video link not found in source."}
+
+    file_id = hashlib.md5(f"{video_url}_{target_quality}".encode()).hexdigest()
     final_video = os.path.join(TEMP_DIR, f"{file_id}.mp4")
+    
+    # Cache Check
+    if os.path.exists(final_video):
+        os.utime(final_video, None)
+        return {"success": True, "method": "Cache", "video_url": f"get_file?id={file_id}.mp4", "size": os.path.getsize(final_video)}
 
+    # Processing
     try:
         v_tmp = os.path.join(TEMP_DIR, f"{uuid.uuid4()}_v.mp4")
         a_tmp = os.path.join(TEMP_DIR, f"{uuid.uuid4()}_a.mp3")
-
-        headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'}
-        
-        # Download Video
-        v_res = requests.get(video_url, headers=headers, timeout=60)
-        with open(v_tmp, 'wb') as f: f.write(v_res.content)
         
-        # Download Audio
-        has_audio = False
+        # Download streams
+        headers = {'User-Agent': 'Mozilla/5.0'}
+        with open(v_tmp, 'wb') as f: f.write(requests.get(video_url, headers=headers).content)
         if audio_url:
-            a_res = requests.get(audio_url, headers=headers, timeout=60)
-            if a_res.status_code == 200:
-                with open(a_tmp, 'wb') as f: f.write(a_res.content)
-                if os.path.getsize(a_tmp) > 1000:
-                    has_audio = True
+            with open(a_tmp, 'wb') as f: f.write(requests.get(audio_url, headers=headers).content)
 
-        # FFmpeg Merging (Ultra High Quality Settings)
+        # Build FFmpeg Command (Force H.264 & AAC)
         cmd = ['ffmpeg', '-y', '-i', v_tmp]
-        if has_audio:
+        if os.path.exists(a_tmp) and os.path.getsize(a_tmp) > 1000:
             cmd += ['-i', a_tmp, '-map', '0:v:0', '-map', '1:a:0']
         
-        # -crf 18 က Original နဲ့ တစ်ထပ်တည်း ရုပ်ထွက်ကို ပေးပါတယ်
-        cmd += [
-            '-c:v', 'libx264', '-crf', '18', '-preset', 'superfast', 
-            '-c:a', 'aac', '-b:a', '192k', '-movflags', '+faststart', final_video
-        ]
+        # Quality Scaling if needed
+        if target_quality != "original":
+            cmd += ['-vf', f"scale=w='trunc(oh*a/2)*2':h={target_quality}"]
+            
+        cmd += ['-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '23', '-c:a', 'aac', '-b:a', '128k', '-movflags', '+faststart', final_video]
         
         subprocess.run(cmd, check=True, capture_output=True)
 
         if os.path.exists(v_tmp): os.remove(v_tmp)
         if os.path.exists(a_tmp): os.remove(a_tmp)
 
-        space_id = os.getenv('SPACE_ID', '').replace("/", "-")
         return {
-            "success": True, 
-            "video_url": f"https://{space_id}.hf.space/get_file?id={file_id}.mp4",
+            "success": True,
+            "method": "Bumiz Engine",
+            "video_url": f"get_file?id={file_id}.mp4",
             "size": os.path.getsize(final_video)
         }
     except Exception as e:
+
+
         return {"success": False, "error": str(e)}
 
 @app.get("/get_file")
 async def get_file(id: str):
     file_path = os.path.join(TEMP_DIR, id)
     if os.path.exists(file_path):
-        return FileResponse(file_path, media_type="video/mp4", filename=f"fb_hq_download.mp4", background=BackgroundTask(cleanup, file_path))
-    raise HTTPException(status_code=404)
-
-@app.get("/download")
-async def download_public(url: str = Query(...)):
-    # Public link fallback
-    return {"success": False, "error": "Please use Private Mode for HQ results."}
+        return FileResponse(file_path, media_type="video/mp4", filename=f"fb_bumiz_{id}", background=BackgroundTask(cleanup, file_path))
+    raise HTTPException(status_code=404)