Update app.py

app.py

@@ -6,6 +6,7 @@ import json
 import requests
 from datetime import datetime, timedelta, timezone
 from flask import Flask, request, jsonify, Response
+from huggingface_hub import InferenceClient
 
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY")
@@ -15,6 +16,7 @@ GROQ_API_KEY_1 = os.getenv("GROQ_API_KEY_1")
 GROQ_API_KEY_2 = os.getenv("GROQ_API_KEY_2")
 GROQ_API_KEY_3 = os.getenv("GROQ_API_KEY_3")
 SERPAPI_KEY = os.getenv("SERPAPI_KEY")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 # ==== URL ====
 GROQ_URL_CHAT = "https://api.groq.com/openai/v1/chat/completions"
@@ -190,7 +192,30 @@ def stream_chat(prompt: str, history=None):
                     yield delta
             except:
                 continue
+# =========================
+# 🎬 Text-to-Video (HF + fal-ai)
+# =========================
+def generate_video(prompt: str, model="Wan-AI/Wan2.2-T2V-A14B-Diffusers"):
+    try:
+        print(f"[VIDEO] 🎬 Generating video for prompt: {prompt}")
+        client = InferenceClient(provider="fal-ai", api_key=HF_TOKEN)
+
+        video_bytes = client.text_to_video(prompt=prompt, model=model)
 
+        if not video_bytes:
+            print("[VIDEO] ❌ Empty response from HF model")
+            return None
+
+        file_name = f"/tmp/video_{int(time.time())}.mp4"
+        with open(file_name, "wb") as f:
+            f.write(video_bytes)
+        print(f"[VIDEO] ✅ Video saved to {file_name}")
+        return file_name
+
+    except Exception as e:
+        print(f"[VIDEO] ❌ Error: {e}")
+        return None
+        
 # =========================
 # 🚀 Chat Endpoint (Text + Voice)
 # =========================
@@ -245,7 +270,37 @@ def chat():
             yield chunk
 
     return Response(generate(), mimetype="text/plain")
+# =========================
+# 🎞️ Video Endpoint
+# =========================
+@app.route("/video", methods=["POST"])
+def video():
+    try:
+        data = request.get_json(force=True)
+        prompt = data.get("prompt", "")
+        if not prompt:
+            return jsonify({"error": "Missing prompt"}), 400
+
+        # Jalankan text-to-video
+        video_path = generate_video(prompt)
+
+        if not video_path or not os.path.exists(video_path):
+            return jsonify({"error": "Failed to generate video"}), 500
 
+        # Encode ke base64 biar mudah dikirim via API
+        with open(video_path, "rb") as f:
+            video_b64 = base64.b64encode(f.read()).decode()
+
+        print(f"[VIDEO] 🎥 Returning base64 video ({len(video_b64)} chars)")
+
+        return jsonify({
+            "prompt": prompt,
+            "video_base64": f"data:video/mp4;base64,{video_b64}"
+        })
+
+    except Exception as e:
+        print(f"[VIDEO] ❌ Exception in endpoint: {e}")
+        return jsonify({"error": str(e)}), 500
 # =========================
 # ▶️ Run
 # =========================