Update main.py

main.py

@@ -21,15 +21,14 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-TEMP_DIR = "/tmp/facebook_downloads"
+TEMP_DIR = "/tmp/fb_cache"
 if not os.path.exists(TEMP_DIR):
     os.makedirs(TEMP_DIR)
 
-MAX_FILE_AGE = 1800 
-
 # --- CONFIGURATION ---
-RAPID_API_KEY = "76af57b863msh20fecf9aaa3b6c6p1bbc52jsn3b00ca30e34f" # <--- BRO ရဲ့ KEY ကို ဒီမှာပြန်ထည့်ပါ
-RAPID_API_HOST = "facebook-reel-and-video-downloader.p.rapidapi.com"
+RAPID_API_KEY = "3a2dadb7dcms..." # <--- ဘရိုရဲ့ KEY အရှည်ကြီးကို ဒီမှာထည့်ပါ
+RAPID_API_HOST = "facebook-media-downloader1.p.rapidapi.com"
+MAX_FILE_AGE = 1800 
 
 def get_url_hash(url: str):
     return hashlib.md5(url.encode()).hexdigest()
@@ -41,18 +40,9 @@ def remove_stale_files():
         if os.path.isfile(file_path) and os.path.getmtime(file_path) < now - MAX_FILE_AGE:
             os.remove(file_path)
 
-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 API: Strict Failure Protection Active"}
+    return {"message": "Bumiz FB API: JSON POST Method Active"}
 
 @app.get("/download")
 async def download_fb_video(background_tasks: BackgroundTasks, url: str = Query(...)):
@@ -67,80 +57,73 @@ async def download_fb_video(background_tasks: BackgroundTasks, url: str = Query(
 
     # 1. CACHE CHECK
     if os.path.exists(video_path):
-        logs.append("Cache hit! Serving existing file.")
+        logs.append("Cache Hit: Found existing file on server.")
         return {
             "success": True, 
             "method": "Bumiz Cache", 
-            "video_url": f"{base_url}{file_id}.mp4", 
-            "music_url": f"{base_url}{file_id}.mp3" if os.path.exists(audio_path) else None, 
+            "video_url": f"{base_url}{file_id}.mp4",
+            "music_url": f"{base_url}{file_id}.mp3" if os.path.exists(audio_path) else None,
             "logs": logs
         }
 
-    # 2. PHASE 1: yt-dlp with Fail-safe
-    temp_raw = os.path.join(TEMP_DIR, f"{file_id}_raw.mp4")
-    logs.append("Phase 1: Attempting yt-dlp (HQ Encoded Mode)...")
-    
+    # 2. PHASE 1: RapidAPI (POST Method with JSON Body)
+    # ဘရိုပြောတဲ့ --data '{"url":"..."}' ပုံစံအတိုင်း ခေါ်မှာဖြစ်ပါတယ်
+    logs.append(f"Phase 1: Requesting RapidAPI via POST...")
     try:
-        ydl_opts = {
-            'socket_timeout': 10,
-            'format': 'best',
-            'outtmpl': temp_raw,
-            'merge_output_format': 'mp4',
-            'quiet': True,
-            'no_warnings': True,
-            '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'
+        api_url = f"https://{RAPID_API_HOST}/" # Endpoint URL ကို သတိပြုပါ
+        payload = {"url": url}
+        headers = {
+            "content-type": "application/json",
+            "X-RapidAPI-Key": RAPID_API_KEY,
+            "X-RapidAPI-Host": RAPID_API_HOST
         }
         
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-        
-        if os.path.exists(temp_raw):
-            streams = get_ffprobe_info(temp_raw)
-            if 'video' in streams:
-                logs.append("yt-dlp successful. Encoding to H.264/AAC...")
-                # Video Encoding
-                subprocess.run([
-                    'ffmpeg', '-y', '-i', temp_raw,
-                    '-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '28', 
-                    '-c:a', 'aac', '-b:a', '128k',
-                    '-movflags', '+faststart', video_path
-                ], check=True)
-                # MP3 Extraction
-                subprocess.run(['ffmpeg', '-y', '-i', temp_raw, '-vn', '-acodec', 'libmp3lame', '-ab', '192k', audio_path], check=True)
-                
-                os.remove(temp_raw)
-                return {"success": True, "method": "yt-dlp (HQ Encoded)", "video_url": f"{base_url}{file_id}.mp4", "music_url": f"{base_url}{file_id}.mp3", "logs": logs}
-    
-    except Exception as e:
-        logs.append(f"yt-dlp skipped due to error: {str(e)}")
-        if os.path.exists(temp_raw): os.remove(temp_raw)
-
-    # 3. PHASE 2: RapidAPI Fallback (Last Resort)
-    logs.append("Phase 2: Switching to RapidAPI for guaranteed success...")
-    try:
-        api_url = f"https://{RAPID_API_HOST}/index"
-        headers = {"X-RapidAPI-Key": RAPID_API_KEY, "X-RapidAPI-Host": RAPID_API_HOST}
-        response = requests.get(api_url, headers=headers, params={"url": url}, timeout=15)
+        response = requests.post(api_url, json=payload, headers=headers, timeout=15)
         data = response.json()
         
-        if data and "video" in data:
-            logs.append("RapidAPI success! Returning HD link.")
+        # API ကနေ ဘာတွေပြန်ပေးလဲဆိုတာ သိရအောင် Log ထုတ်ကြည့်ခြင်း
+        logs.append(f"API Response Received.")
+
+        # API JSON Structure အလိုက် လင့်ခ်များကို ဆွဲထုတ်ခြင်း (AssadAzeem API Format)
+        video_url = data.get("hd") or data.get("sd") or data.get("url")
+        audio_url = data.get("audio") or data.get("mp3")
+
+        if video_url:
+            logs.append("API Success: High Quality link found.")
             return {
-                "success": True, 
-                "method": "RapidAPI", 
-                "video_url": data["video"][0], 
-                "music_url": data["music"][0], 
+                "success": True,
+                "method": "RapidAPI (POST)",
+                "video_url": video_url,
+                "music_url": audio_url,
                 "logs": logs
             }
+        else:
+            logs.append(f"API Error: No video link in response. JSON: {str(data)[:100]}")
     except Exception as e:
-        logs.append(f"RapidAPI also failed: {str(e)}")
-    
-    return {"success": False, "error": "All methods failed to process this video.", "logs": logs}
+        logs.append(f"API Connection Failed: {str(e)}")
+
+    # 3. PHASE 2: yt-dlp (Fallback)
+    logs.append("Phase 2: yt-dlp Fallback started...")
+    temp_raw = os.path.join(TEMP_DIR, f"{file_id}_raw.mp4")
+    try:
+        ydl_opts = {'format': 'best', 'outtmpl': temp_raw, 'quiet': True, 'socket_timeout': 10}
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        
+        if os.path.exists(temp_raw):
+            logs.append("yt-dlp success! Encoding file...")
+            subprocess.run(['ffmpeg', '-y', '-i', temp_raw, '-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '28', '-c:a', 'aac', '-b:a', '128k', '-movflags', '+faststart', video_path], check=True)
+            os.remove(temp_raw)
+            return {"success": True, "method": "yt-dlp (Fallback)", "video_url": f"{base_url}{file_id}.mp4", "logs": logs}
+    except Exception as e:
+        logs.append(f"yt-dlp failed: {str(e)}")
+
+    return {"success": False, "error": "Could not download this Facebook video.", "logs": logs}
 
 @app.get("/get_file")
 async def get_file(id: str):
     file_path = os.path.join(TEMP_DIR, id)
     if os.path.exists(file_path):
         media_type = "video/mp4" if id.endswith(".mp4") else "audio/mpeg"
-        return FileResponse(file_path, media_type=media_type, filename=f"fb_download_{id}")
-    raise HTTPException(status_code=404, detail="File expired or not found")
+        return FileResponse(file_path, media_type=media_type, filename=f"fb_{id}")
+    raise HTTPException(status_code=404, detail="File expired")