Update main.py

main.py

@@ -27,9 +27,7 @@ TEMP_DIR = "/tmp/fb_cache"
 if not os.path.exists(TEMP_DIR):
     os.makedirs(TEMP_DIR)
 
-# --- CONFIGURATION ---
-RAPID_API_KEY = "7...f" # <-- Bro ရဲ့ Key ကို ပြန်ထည့်ပါ
-RAPID_API_HOST = "facebook-media-downloader1.p.rapidapi.com"
+MAX_FILE_AGE = 1800 
 
 def cleanup(file_path: str):
     if os.path.exists(file_path):
@@ -43,129 +41,123 @@ def clean_fb_url(url: str):
     except: pass
     return urllib.parse.unquote(url).replace('&amp;', '&')
 
-def download_temp_file(url: str, suffix: str):
-    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=20, 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
-        if os.path.exists(path): os.remove(path)
-    except Exception as e:
-        print(f"Download failed: {e}")
-        if os.path.exists(path): os.remove(path)
-    return None
-
-def get_stream_type(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 API: Cleaned Error Handling"}
+    return {"message": "Bumiz Facebook Engine: High-Quality Direct Scraper Active"}
 
-@app.post("/download-private")
-async def download_private_fb(background_tasks: BackgroundTasks, payload: dict = Body(...)):
+# --- ၁။ SOURCE CODE ကို ANALYZE လုပ်ပြီး RESOLUTION အားလုံး ရှာဖွေခြင်း ---
+@app.post("/analyze-private")
+async def analyze_private_fb(payload: dict = Body(...)):
     html_source = payload.get("html", "")
-    logs = ["Scraper: Starting..."]
-    
     if not html_source:
-        return {"success": False, "error": "HTML Source is empty", "logs": logs}
+        return {"success": False, "error": "HTML Source is empty"}
 
-    video_url = None
+    found_qualities = []
+    # Video Representations ရှာဖွေခြင်း
+    matches = re.findall(r'"height":(\d+),.*?"base_url":"([^"]+)"', html_source)
+    
+    seen_urls = set()
+    for height, url in matches:
+        h = int(height)
+        u = clean_fb_url(url)
+        if u and u not in seen_urls and "fbcdn.net" in u:
+            found_qualities.append({
+                "label": f"{h}p Quality",
+                "url": u,
+                "height": h,
+                "size_mb": 0,
+                "type": "hd" if h >= 720 else "sd"
+            })
+            seen_urls.add(u)
+
+    found_qualities.sort(key=lambda x: x['height'], reverse=True)
+
+    # Audio Link ရှာဖွေခြင်း
     audio_url = None
+    a_match = re.search(r'"mime_type":"audio/[^"]+","base_url":"([^"]+)"', html_source)
+    if a_match:
+        audio_url = clean_fb_url(a_match.group(1))
+
+    if not found_qualities:
+        return {"success": False, "error": "No videos found in source code."}
 
-    v_patterns = [
-        r'"browser_native_hd_url":"([^"]+)"',
-        r'"playable_url_quality_hd":"([^"]+)"',
-        r'"browser_native_sd_url":"([^"]+)"',
-        r'"playable_url":"([^"]+)"',
-        r'"base_url":"([^"]+)"'
-    ]
-    for p in v_patterns:
-        match = re.search(p, html_source)
-        if match:
-            url = clean_fb_url(match.group(1))
-            if url and "fbcdn.net" in url and "/v/" in url:
-                video_url = url
-                logs.append(f"Found Video: {url[:80]}...")
-                break
-
-    a_patterns = [
-        r'"mime_type":"audio/[^"]+".*?"base_url":"([^"]+)"',
-        r'"audio_channel_configuration".*?"base_url":"([^"]+)"'
-    ]
-    for p in a_patterns:
-        match = re.search(p, html_source)
-        if match:
-            url = clean_fb_url(match.group(1))
-            if url and "fbcdn.net" in url:
-                audio_url = url
-                logs.append(f"Found Audio: {url[:80]}...")
-                break
-
-    if not video_url:
-        return {"success": False, "error": "Video link not found", "logs": logs}
-
-    logs.append("Downloading streams...")
+    return {"success": True, "qualities": found_qualities, "audio_url": audio_url}
+
+# --- ၂။ ရွေးချယ်လိုက်သော QUALITY ကို အကြည်ဆုံး ဖြစ်အောင် PROCESSING လုပ်ခြင်း ---
+@app.post("/process-private")
+async def process_private_fb(background_tasks: BackgroundTasks, payload: dict = Body(...)):
+    # request လာတိုင်း အဟော��်းတွေကို ရှင်းမယ်
+    background_tasks.add_task(lambda: [os.remove(os.path.join(TEMP_DIR, f)) for f in os.listdir(TEMP_DIR) if time.time() - os.stat(os.path.join(TEMP_DIR, f)).st_mtime > MAX_FILE_AGE])
     
+    video_url = payload.get("video_url")
+    audio_url = payload.get("audio_url")
+    compress_mode = payload.get("compress_mode")
+
     file_id = str(uuid.uuid4())
-    video_path = os.path.join(TEMP_DIR, f"{file_id}.mp4")
-    v_tmp = None
-    a_tmp = None
+    final_video_path = os.path.join(TEMP_DIR, f"{file_id}.mp4") # NameError မဖြစ်အောင် အပေါ်မှာ ကြိုသတ်မှတ်သည်
 
     try:
-        v_tmp = download_temp_file(video_url, ".mp4")
-        if not v_tmp: raise Exception("Video download failed")
+        v_tmp = os.path.join(TEMP_DIR, f"{file_id}_v.mp4")
+        a_tmp = os.path.join(TEMP_DIR, f"{file_id}_a.mp3")
+
+        headers = {'User-Agent': 'Mozilla/5.0'}
+        with open(v_tmp, 'wb') as f: f.write(requests.get(video_url, headers=headers, timeout=60).content)
         
-        a_tmp = download_temp_file(audio_url, ".mp3") if audio_url else None
-        if audio_url and not a_tmp: logs.append("Warning: Audio download failed. Continuing without audio.")
+        has_audio = False
+        if audio_url:
+            with open(a_tmp, 'wb') as f: f.write(requests.get(audio_url, headers=headers, timeout=60).content)
+            if os.path.exists(a_tmp) and os.path.getsize(a_tmp) > 1000:
+                has_audio = True
 
-        # ENCODING
-        logs.append("Starting HQ Encoding...")
+        # FFmpeg Command Build
         cmd = ['ffmpeg', '-y', '-i', v_tmp]
-        if a_tmp:
+        if has_audio:
             cmd += ['-i', a_tmp, '-map', '0:v:0', '-map', '1:a:0']
         
-        cmd += ['-c:v', 'libx264', '-preset', 'fast', '-crf', '23', '-c:a', 'aac', '-b:a', '128k', '-movflags', '+faststart', final_video]
-        
-        # Log တွေကို ထုတ်ကြည့်ဖို့ သေချာထည့်ထားပါ
-        result = subprocess.run(cmd, capture_output=True, text=True)
-        logs.append(f"FFmpeg Output: {result.stdout} {result.stderr}")
+        if compress_mode and compress_mode != "original":
+            cmd += ['-vf', f"scale=w='trunc(oh*a/2)*2':h={compress_mode}"]
+
+        # Force High Quality Settings
+        cmd += ['-c:v', 'libx264', '-crf', '20', '-preset', 'superfast', '-c:a', 'aac', '-b:a', '128k', '-movflags', '+faststart', final_video_path]
         
-        # Processing အောင်မြင်လား စစ်ဆေးခြင်း
-        if not os.path.exists(final_video) or os.path.getsize(final_video) < 1000:
-            raise Exception("Encoding failed: FFmpeg returned an empty file.")
+        subprocess.run(cmd, check=True, capture_output=True)
 
-        if v_tmp: os.remove(v_tmp)
-        if a_tmp: os.remove(a_tmp)
+        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,
-            "method": "Private (Full Processing)",
-            "video_url": f"https://{space_id}.hf.space/get_file?id={file_id}.mp4",
-            "logs": logs
-        }
+        return {"success": True, "video_url": f"https://{space_id}.hf.space/get_file?id={file_id}.mp4"}
 
     except Exception as e:
-        if v_tmp: os.remove(v_tmp)
-        if a_tmp: os.remove(a_tmp)
-        logs.append(f"CRITICAL ERROR: {str(e)}")
-        return {"success": False, "error": str(e), "logs": logs}
+        return {"success": False, "error": str(e)}
+
+# --- ၃။ PUBLIC DOWNLOAD LOGIC (No RapidAPI) ---
+@app.get("/download")
+async def download_public(url: str = Query(...)):
+    file_id = hashlib.md5(url.encode()).hexdigest()
+    video_path = os.path.join(TEMP_DIR, f"{file_id}.mp4")
+    
+    if os.path.exists(video_path):
+        space_id = os.getenv('SPACE_ID', '').replace("/", "-")
+        return {"success": True, "video_url": f"get_file?id={file_id}.mp4"}
+
+    temp_raw = os.path.join(TEMP_DIR, f"{file_id}_raw.mp4")
+    try:
+        ydl_opts = {'format': 'best', 'outtmpl': temp_raw, 'quiet': True, 'socket_timeout': 20}
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        
+        if os.path.exists(temp_raw):
+            subprocess.run(['ffmpeg', '-y', '-i', temp_raw, '-c:v', 'libx264', '-crf', '22', '-preset', 'ultrafast', '-c:a', 'aac', '-movflags', '+faststart', video_path], check=True)
+            os.remove(temp_raw)
+            space_id = os.getenv('SPACE_ID', '').replace("/", "-")
+            return {"success": True, "video_url": f"get_file?id={file_id}.mp4"}
+    except: pass
+    return {"success": False, "error": "Public failed. Try Private Mode."}
 
 @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_private_{id}", background=BackgroundTask(cleanup, file_path))
-    raise HTTPException(status_code=404, detail="File expired")
+        return FileResponse(file_path, media_type="video/mp4", filename=f"bumiz_fb_{id}", background=BackgroundTask(cleanup, file_path))
+    raise HTTPException(status_code=404)