import requests
import gradio as gr
from dotenv import load_dotenv
import os

load_dotenv()

def generate_upf_code(message, history):
    api_url = "https://api.dify.ai/v1/chat-messages"
    api_key = os.getenv("API_KEY")

    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }

    payload = {
        "inputs": {},
        "query": message,
        "response_mode": "blocking",
        "conversation_id": "",
        "user": "abc-123"
    }

    response = requests.post(api_url, headers=headers, json=payload)

    if not response.ok:
        raise Exception(f"API Error: {response.status_code} {response.reason}")

    data = response.json()
    answer = data["answer"]
    return answer

gr.ChatInterface(
    fn=generate_upf_code, 
    type="messages",
    title="UPF Code Generator",
    description="Ask me to generate UPF code for your design constraints.",
    theme="ocean"
).launch()