fix-translate

app.py

@@ -0,0 +1,94 @@
+import gradio as gr
+import requests
+import json
+
+# List of supported languages
+LANGUAGES = [
+    "Assamese (asm_Beng)", "Kashmiri (Arabic) (kas_Arab)", "Punjabi (pan_Guru)",
+    "Bengali (ben_Beng)", "Kashmiri (Devanagari) (kas_Deva)", "Sanskrit (san_Deva)",
+    "Bodo (brx_Deva)", "Maithili (mai_Deva)", "Santali (sat_Olck)",
+    "Dogri (doi_Deva)", "Malayalam (mal_Mlym)", "Sindhi (Arabic) (snd_Arab)",
+    "English (eng_Latn)", "Marathi (mar_Deva)", "Sindhi (Devanagari) (snd_Deva)",
+    "Konkani (gom_Deva)", "Manipuri (Bengali) (mni_Beng)", "Tamil (tam_Taml)",
+    "Gujarati (guj_Gujr)", "Manipuri (Meitei) (mni_Mtei)", "Telugu (tel_Telu)",
+    "Hindi (hin_Deva)", "Nepali (npi_Deva)", "Urdu (urd_Arab)",
+    "Kannada (kan_Knda)", "Odia (ory_Orya)"
+]
+
+# Function to extract language code from selection
+def get_lang_code(lang_string):
+    return lang_string.split("(")[-1].rstrip(")")
+
+def translate_api(sentences, src_lang, tgt_lang):
+    url = "https://slabstech-dhwani-server-workshop.hf.space/v1/translate"
+    
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/json"
+    }
+    
+    # Convert sentences string to list if it's a string
+    if isinstance(sentences, str):
+        try:
+            sentences_list = json.loads(sentences)
+        except json.JSONDecodeError:
+            sentences_list = [sentences]
+    else:
+        sentences_list = sentences
+    
+    payload = {
+        "sentences": sentences_list,
+        "src_lang": get_lang_code(src_lang),
+        "tgt_lang": get_lang_code(tgt_lang)
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.HTTPError as e:
+        return {"error": f"HTTP Error: {str(e)}"}
+    except requests.exceptions.RequestException as e:
+        return {"error": f"Request Error: {str(e)}"}
+
+# Create Gradio interface
+with gr.Blocks(title="Translation API Interface") as demo:
+    gr.Markdown("# Translation API Interface")
+    gr.Markdown("Enter sentences and select languages to translate.")
+    
+    with gr.Row():
+        with gr.Column():
+            # Input components
+            sentences_input = gr.Textbox(
+                label="Sentences",
+                placeholder='Enter sentences as JSON array or single sentence (e.g., ["Hello", "Good morning"] or "Hello")',
+                lines=3,
+                value='["Hi"]'
+            )
+            src_lang_input = gr.Dropdown(
+                label="Source Language",
+                choices=LANGUAGES,
+                value="English (eng_Latn)"
+            )
+            tgt_lang_input = gr.Dropdown(
+                label="Target Language",
+                choices=LANGUAGES,
+                value="Kannada (kan_Knda)"
+            )
+            
+            submit_btn = gr.Button("Translate")
+        
+        with gr.Column():
+            # Output component
+            output = gr.JSON(label="Translation Response")
+    
+    # Connect the button click to the API function
+    submit_btn.click(
+        fn=translate_api,
+        inputs=[sentences_input, src_lang_input, tgt_lang_input],
+        outputs=output
+    )
+
+# Launch the interface
+if __name__ == "__main__":
+    demo.launch()