Update app.py

app.py

@@ -4,41 +4,39 @@ import os
 
 # Retrieve secrets
 API_URL = os.environ.get("API_URL")
-HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# Hugging Face Inference API details
-headers = {
-    "Accept": "application/json",
-    "Authorization": f"Bearer {HF_TOKEN}",
-    "Content-Type": "application/json"
-}
+import gradio as gr
+import requests
+import os
+
+headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-def translate(text, is_english_to_welsh):
-    source_lang = "English" if is_english_to_welsh else "Welsh"
-    target_lang = "Welsh" if is_english_to_welsh else "English"
+def translate(text, source_lang, target_lang):
+    if source_lang == target_lang:
+        return text
     
     instruction = f"Translate the text from {source_lang} to {target_lang}"
     
     input_text = f"""### Instruction: {instruction}
+
 ### Input: {text}
+
 ### Response:
 """
     
-    # Generate the translation using the Hugging Face Inference API
     output = query({
         "inputs": input_text,
         "parameters": {
-            "return_text": False,
             "return_full_text": False
         }
     })
     
-    # Extract the translated text from the API response
-    translated_text = output[0]['generated_text']
+    # Extract the translated text from the model output
+    translated_text = output[0]['generated_text'].strip()
     
     return translated_text
 
@@ -47,14 +45,15 @@ iface = gr.Interface(
     fn=translate,
     inputs=[
         gr.Textbox(label="Enter text to translate"),
-        gr.Checkbox(label="English to Welsh", value=True),
+        gr.Radio(["English", "Welsh"], label="Source Language", value="English"),
+        gr.Radio(["English", "Welsh"], label="Target Language", value="Welsh"),
     ],
     outputs=gr.Textbox(label="Translated Text"),
     title="English-Welsh Translator",
-    description="Translate text between English and Welsh using a custom language model via Hugging Face Inference API.",
+    description="Translate text between English and Welsh using a Hugging Face Inference Endpoint.",
     examples=[
-        ["Hello, how are you?", True],
-        ["Bore da!", False],
+        ["Hello, how are you?", "English", "Welsh"],
+        ["Bore da!", "Welsh", "English"],
     ],
 )