import requests
import json
import gradio as gr
import os

# Define the function to call the API
def translate(text):
    url = os.getenv("TRANSLATION_URL")
    payload = json.dumps({
        "text": text
    })
    headers = {
        'Content-Type': 'application/json'
    }
    
    # Make the API request
    response = requests.request("POST", url, headers=headers, data=payload)
    
    # Parse the response and extract the translation
    response_data = response.json()  # Parse the JSON response
    translation = response_data.get("translation", "Translation not found")  # Get the translation
    
    return translation

# Create the Gradio interface with a placeholder in the text box
textbox = gr.inputs.Textbox(placeholder="Enter english text to translate...")

demo = gr.Interface(fn=translate, inputs=textbox, outputs="text")

# Launch the Gradio app
demo.launch()