Spaces:
Sleeping
Sleeping
File size: 4,444 Bytes
4e1fde2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
import gradio as gr
import requests
import json
import os
import dwani
import logging
# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
# Configure dwani API settings
dwani.api_key = os.getenv("DWANI_API_KEY")
dwani.api_base = os.getenv("DWANI_API_BASE_URL")
# Log API configuration for debugging
logger.debug("DWANI_API_KEY: %s", "Set" if dwani.api_key else "Not set")
logger.debug("DWANI_API_BASE_URL: %s", dwani.api_base)
# List of supported languages
LANGUAGES = [
"Assamese", "Punjabi",
"Bengali", "Malayalam ",
"English", "Marathi ", "Tamil",
"Gujarati", "Telugu ",
"Hindi", "Kannada", "Odia "
]
def translate_api(sentences, src_lang, tgt_lang):
logger.debug("Received inputs - Sentences: %s, Source Lang: %s, Target Lang: %s", sentences, src_lang, tgt_lang)
# Convert simple string input to list
try:
if isinstance(sentences, str):
# Split on commas and strip whitespace
sentences = [s.strip() for s in sentences.split(",") if s.strip()]
elif isinstance(sentences, list):
# Ensure all elements are strings and non-empty
sentences = [s.strip() for s in sentences if isinstance(s, str) and s.strip()]
else:
logger.error("Invalid input type for sentences: %s", type(sentences))
return {"error": "Invalid input: sentences must be a string or list of strings"}
except Exception as e:
logger.error("Error processing sentences input: %s", str(e))
return {"error": f"Error processing input: {str(e)}"}
# Validate sentences content
if not sentences:
logger.error("No valid sentences provided: %s", sentences)
return {"error": "Please provide at least one non-empty sentence"}
# Extract language codes
src_lang_code = (src_lang)
tgt_lang_code = (tgt_lang)
if not src_lang_code or not tgt_lang_code:
logger.error("Invalid language codes - Source: %s, Target: %s", src_lang_code, tgt_lang_code)
return {"error": "Invalid source or target language selection"}
logger.debug("Calling API with sentences: %s, src_lang: %s, tgt_lang: %s", sentences, src_lang_code, tgt_lang_code)
# Call the API
try:
result = dwani.Translate.run_translate(sentences=sentences, src_lang=src_lang_code, tgt_lang=tgt_lang_code)
logger.debug("API response: %s", result)
return result
except dwani.exceptions.DhwaniAPIError as e:
logger.error("Dhwani API error: %s", str(e))
return {"error": f"API error: {str(e)}"}
except Exception as e:
logger.error("Unexpected error: %s", str(e))
return {"error": f"Unexpected error: {str(e)}"}
# Create Gradio interface
with gr.Blocks(title="Translation API Interface") as demo:
gr.Markdown("# Translation API Interface")
gr.Markdown("Enter one or more sentences (separated by commas) and select languages to translate.")
with gr.Row():
with gr.Column():
# Input components
sentences_input = gr.Textbox(
label="Sentences",
placeholder="Enter your sentence (e.g., Hello, Good morning)",
lines=3,
value="Hi"
)
src_lang_input = gr.Dropdown(
label="Source Language",
choices=LANGUAGES,
value="English"
)
tgt_lang_input = gr.Dropdown(
label="Target Language",
choices=LANGUAGES,
value="Kannada"
)
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__":
# Test API configuration
if not dwani.api_key or not dwani.api_base:
logger.error("API key or base URL not set. Please set DWANI_API_KEY and DWANI_API_BASE_URL environment variables.")
print("Error: Please set DWANI_API_KEY and DWANI_API_BASE_URL environment variables.")
else:
logger.debug("Starting Gradio interface...")
demo.launch() |