Create facebook_main.py

facebook_main.py

@@ -0,0 +1,125 @@
+import yt_dlp
+from fastapi import FastAPI, HTTPException, Query, BackgroundTasks
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse
+import subprocess
+import os
+import uuid
+import json
+import requests
+import time
+import hashlib
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+TEMP_DIR = "/tmp/facebook_downloads"
+if not os.path.exists(TEMP_DIR):
+    os.makedirs(TEMP_DIR)
+
+# --- CONFIGURATION ---
+# Facebook အတွက် RapidAPI ရှိလျှင် Key ထည့်ပါ (မရှိသေးရင် yt-dlp နဲ့ပဲ အရင်အလုပ်လုပ်ပါမယ်)
+RAPID_API_KEY = "YOUR_RAPID_API_KEY" 
+RAPID_API_HOST = "facebook-reel-and-video-downloader.p.rapidapi.com"
+MAX_FILE_AGE = 1800 # ၃၀ မိနစ်
+
+def get_url_hash(url: str):
+    return hashlib.md5(url.encode()).hexdigest()
+
+def remove_stale_files():
+    now = time.time()
+    for f in os.listdir(TEMP_DIR):
+        file_path = os.path.join(TEMP_DIR, f)
+        if os.path.isfile(file_path):
+            if 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 Facebook API: Isolated Micro-service Active"}
+
+@app.get("/download")
+async def download_fb_video(background_tasks: BackgroundTasks, url: str = Query(...)):
+    remove_stale_files()
+    logs = []
+    file_id = get_url_hash(url)
+    video_path = os.path.join(TEMP_DIR, f"{file_id}.mp4")
+    audio_path = os.path.join(TEMP_DIR, f"{file_id}.mp3")
+    
+    space_id = os.getenv('SPACE_ID', '').replace("/", "-")
+    base_url = f"https://{space_id}.hf.space/get_file?id="
+
+    # ၁။ CACHE CHECK
+    if os.path.exists(video_path):
+        os.utime(video_path, None)
+        logs.append("Bumiz Cache: Found existing file.")
+        return {
+            "success": True, 
+            "method": "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,
+            "logs": logs
+        }
+
+    # ၂။ Phase 1: yt-dlp
+    logs.append("Phase 1: Attempting yt-dlp (Facebook Optimized)...")
+    temp_raw = os.path.join(TEMP_DIR, f"{file_id}_raw.mp4")
+    try:
+        ydl_opts = {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+            'outtmpl': temp_raw,
+            'quiet': True,
+            'no_warnings': True
+        }
+        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("FFmpeg: Forcing H.264 and AAC (128k)...")
+                # Video Processing
+                subprocess.run(['ffmpeg', '-y', '-i', temp_raw, '-c:v', 'libx264', '-preset', 'superfast', '-crf', '22', '-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 failed: {str(e)}")
+        if os.path.exists(temp_raw): os.remove(temp_raw)
+
+    # ၃။ Phase 2: RapidAPI (Fallback)
+    logs.append("Phase 2: Switching to RapidAPI...")
+    # မှတ်ချက် - FB API key ရှိမှသာ အလုပ်လုပ်ပါမည်
+    return {"success": False, "error": "Could not process 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):
+        os.utime(file_path, None)
+        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")