import gradio as gr
import requests
import uuid

session_id = str(uuid.uuid4())

def generate_text(prompt):
    url = "https://mgpzj8t1f2.execute-api.us-east-1.amazonaws.com/DEV/generate"
    headers = {"Content-Type": "application/json"}
    data = {"prompt": prompt, "sessionId": session_id}

    response = requests.post(url, headers=headers, json=data)
    response.raise_for_status()  # Ensure we raise an error for bad responses
    
    return response.text.replace('"', '').replace("\\n", ' ').replace('/',' ').replace('\\', ' ')

iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
    outputs=gr.Textbox(label="Generated Text"),
    live=False,
    allow_flagging="never"
)

iface.launch()