import os
from fastapi import FastAPI
import requests
import uvicorn

app = FastAPI()

# API Key from Hugging Face Secrets
API_KEY = os.getenv("HF_API_KEY")  
API_URL = "https://api-inference.huggingface.co/models/damo-vilab/modelscope-text-to-video-synthesis"
HEADERS = {"Authorization": f"Bearer {API_KEY}"}

@app.get("/")
def home():
    return {"message": "Welcome to AI Video Generator!"}

@app.get("/generate_video/")
def generate_video(prompt: str):
    response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt})
    return response.json()

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)