import streamlit as st
import os
import requests
import io
from PIL import Image

st.title("Cinematic Text-to-Image Generation")
st.write("Enter text and generate cinematic photo. Optional tag is POV")
prompt = st.text_input("Enter prompt")
hf_token = os.environ.get("api_token")
headers = {"Authorization": f"Bearer {hf_token}"}
API_URL = "https://api-inference.huggingface.co/models/TheLastBen/Filmic"

def query(payload):
  response = requests.post(API_URL, headers=headers, json=payload)
  return response.content
    
def generate_image(input_text):
    image_bytes = query({
        "inputs": input_text,
    })
    image = Image.open(io.BytesIO(image_bytes))
    return image

if prompt:
    output = generate_image(prompt)
    st.image(output, caption="Generated image")