Spaces:
Sleeping
Sleeping
File size: 1,254 Bytes
7006220 d53004d a38b4f9 d53004d 7006220 d53004d a38b4f9 d53004d a38b4f9 d53004d 7006220 a38b4f9 d53004d 7006220 d53004d 7006220 d53004d 7006220 d53004d 7006220 d53004d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
from fastapi import APIRouter, HTTPException, Depends, Body
from fastapi.responses import Response
from pydantic import BaseModel
from app.auth.auth import verify_token
import requests
import os
from pydantic import BaseModel
from dotenv import load_dotenv
load_dotenv()
# ✅ Define router
router = APIRouter()
# ✅ Define Request schema
class ImageRequest(BaseModel):
prompt: str
style: str = "default"
UNSPLASH_ACCESS_KEY = os.getenv("UNSPLASH_ACCESS_KEY") # store this in .env
print(f"unsplash key is: {UNSPLASH_ACCESS_KEY}")
# ✅ Endpoint
@router.post("/generate")
def generate_image_file_endpoint(
data: ImageRequest = Body(...),
token: str = Depends(verify_token)
):
query = f"{data.prompt} {data.style}"
url = f"https://api.unsplash.com/photos/random?query={query}&client_id={UNSPLASH_ACCESS_KEY}&orientation=landscape"
try:
r = requests.get(url)
r.raise_for_status()
image_url = r.json()["urls"]["regular"]
img_data = requests.get(image_url).content
return Response(content=img_data, media_type="image/jpeg")
except Exception as e:
print(f"❌ Image fetch failed: {str(e)}")
raise HTTPException(status_code=500, detail="Image generation failed.")
|