Translation_App / app.py
ItqaAkhlaq's picture
Update app.py
88ca7ca verified
Raw
History Blame Contribute Delete
1.61 kB
import gradio as gr
import requests
import os
# Load API key from environment variable
GROQ_API_KEY = os.getenv("Trans_Groq_API_Key")
def translate_text(input_text, target_language):
headers = {
"Authorization": f"Bearer {GROQ_API_KEY}",
"Content-Type": "application/json"
}
prompt = (
f"Detect the language of the following text and translate it accurately to {target_language}:\n\n"
f"Text: \"{input_text}\"\n\n"
f"Translated Text:"
)
data = {
"model": "llama3-8b-8192",
"messages": [
{"role": "user", "content": prompt}
],
"temperature": 0.2,
"max_tokens": 250
}
response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=data)
if response.status_code == 200:
return response.json()['choices'][0]['message']['content'].strip()
else:
return f"Error: {response.status_code}\n{response.text}"
interface = gr.Interface(
fn=translate_text,
inputs=[
gr.Textbox(lines=4, label="Enter Text (Any Language)"),
gr.Dropdown(
choices=[
"English", "Urdu", "Hindi", "Spanish", "French", "German", "Chinese",
"Arabic", "Russian", "Italian", "Japanese", "Korean", "Portuguese", "Turkish"
],
label="Translate To"
)
],
outputs=gr.Textbox(label="Translated Text"),
title="🌐 AI Translator with Groq",
description="Translate text from any language to your selected language using LLaMA 3 via Groq."
)
interface.launch()