import gradio as gr
import requests
import os

# Get Groq API Key from Hugging Face Secret
GROQ_API_KEY = os.environ.get("GROQ_API_KEY")

MODEL_NAME = "llama3-8b-8192"

def translate_text(english_text, target_language):
    url = "https://api.groq.com/openai/v1/chat/completions"

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

    prompt = f"Translate the following English text to {target_language}:\n\n\"{english_text}\""

    data = {
        "model": MODEL_NAME,
        "messages": [
            {"role": "system", "content": "You are a translation assistant."},
            {"role": "user", "content": prompt}
        ],
        "temperature": 0.3
    }

    response = requests.post(url, headers=headers, json=data)

    if response.status_code == 200:
        result = response.json()
        translation = result['choices'][0]['message']['content']
        return translation
    else:
        return f"Error {response.status_code}: {response.text}"

iface = gr.Interface(
    fn=translate_text,
    inputs=[
        gr.Textbox(placeholder="Enter English text here...", label="English Text"),
        gr.Textbox(placeholder="Enter target language (e.g., Urdu, Spanish)...", label="Target Language")
    ],
    outputs=gr.Textbox(label="Translation"),
    title="🌍 English to Any Language Translator",
    description="Translate English text to your desired language using Groq API (LLaMA 3 model)"
)

iface.launch()