Spaces:
Running
Running
| import gradio as gr | |
| import requests | |
| import json | |
| import os | |
| import time | |
| from collections import defaultdict | |
| from PIL import Image | |
| import io | |
| BASE_URL = "https://api.jigsawstack.com/v1" | |
| headers = { | |
| "x-api-key": os.getenv("JIGSAWSTACK_API_KEY") | |
| } | |
| # Rate limiting configuration | |
| request_times = defaultdict(list) | |
| MAX_REQUESTS = 20 # Maximum requests per time window | |
| TIME_WINDOW = 3600 # Time window in seconds (1 hour) | |
| def get_real_ip(request: gr.Request): | |
| """Extract real IP address using x-forwarded-for header or fallback""" | |
| if not request: | |
| return "unknown" | |
| forwarded = request.headers.get("x-forwarded-for") | |
| if forwarded: | |
| ip = forwarded.split(",")[0].strip() # First IP in the list is the client's | |
| else: | |
| ip = request.client.host # fallback | |
| return ip | |
| def check_rate_limit(request: gr.Request): | |
| """Check if the current request exceeds rate limits""" | |
| if not request: | |
| return True, "Rate limit check failed - no request info" | |
| ip = get_real_ip(request) | |
| now = time.time() | |
| # Clean up old timestamps outside the time window | |
| request_times[ip] = [t for t in request_times[ip] if now - t < TIME_WINDOW] | |
| # Check if rate limit exceeded | |
| if len(request_times[ip]) >= MAX_REQUESTS: | |
| time_remaining = int(TIME_WINDOW - (now - request_times[ip][0])) | |
| time_remaining_minutes = round(time_remaining / 60, 1) | |
| time_window_minutes = round(TIME_WINDOW / 60, 1) | |
| return False, f"Rate limit exceeded. You can make {MAX_REQUESTS} requests per {time_window_minutes} minutes. Try again in {time_remaining_minutes} minutes." | |
| # Add current request timestamp | |
| request_times[ip].append(now) | |
| return True, "" | |
| # ----------------- JigsawStack API Wrappers ------------------ | |
| def translate_text(text_input, target_lang, request: gr.Request): | |
| # Check rate limit first | |
| rate_limit_ok, rate_limit_msg = check_rate_limit(request) | |
| if not rate_limit_ok: | |
| return {"error": "Rate limit exceeded", "message": rate_limit_msg} | |
| # Validate required inputs | |
| if not text_input or not text_input.strip(): | |
| return {"error": "Text input is required"} | |
| if not target_lang or not target_lang.strip(): | |
| return {"error": "Target language code is required"} | |
| # Handle both single string and comma-separated multiple phrases | |
| if "," in text_input and not text_input.strip().startswith("["): | |
| text_payload = [t.strip() for t in text_input.split(",") if t.strip()] | |
| else: | |
| text_payload = text_input.strip() | |
| payload = { | |
| "text": text_payload, | |
| "target_language": target_lang.strip() | |
| } | |
| try: | |
| r = requests.post(f"{BASE_URL}/ai/translate", headers=headers, json=payload) | |
| r.raise_for_status() | |
| data = r.json() | |
| if not data.get("success"): | |
| return {"error": "Translation failed", "response": data} | |
| # Return a more comprehensive result matching the API response | |
| result = { | |
| "success": True, | |
| "translated_text": data.get("translated_text") | |
| } | |
| return result | |
| except requests.exceptions.RequestException as req_err: | |
| return {"error": "Request failed", "message": str(req_err)} | |
| except Exception as e: | |
| return {"error": "Unexpected error", "message": str(e)} | |
| # ----------------- Gradio UI ------------------ | |
| with gr.Blocks() as demo: | |
| gr.Markdown(""" | |
| <div style='text-align: center; margin-bottom: 24px;'> | |
| <h1 style='font-size:2.2em; margin-bottom: 0.2em;'>🧩 Text Translation</h1> | |
| <p style='font-size:1.2em; margin-top: 0;'>Translate text from one language to another with support for multiple text formats.</p> | |
| <p style='font-size:1em; margin-top: 0.5em;'>For more details and API usage, see the <a href='https://jigsawstack.com/docs/api-reference/ai/translate/translate' target='_blank'>documentation</a>.</p> | |
| </div> | |
| """) | |
| with gr.Row(): | |
| # Left Column: Inputs and Examples | |
| with gr.Column(scale=2): | |
| gr.Markdown("#### Input") | |
| text_input = gr.Textbox( | |
| label="Text to Translate", | |
| lines=4, | |
| placeholder="Enter text to translate (use commas to separate multiple phrases for batch translation)" | |
| ) | |
| gr.Markdown("#### Translation Options") | |
| target_lang = gr.Textbox( | |
| label="Target Language Code", | |
| placeholder="es (Spanish), fr (French), de (German), etc.", | |
| info="Common codes: es, fr, de, hi, ja, ko, zh, ar, ru, pt" | |
| ) | |
| gr.Markdown("#### Quick Examples") | |
| gr.Markdown("Click any example below to auto-fill the fields:") | |
| with gr.Row(): | |
| example_btn1 = gr.Button("🇪🇸 Hello World → Spanish", size="sm") | |
| example_btn2 = gr.Button("🇫🇷 Good morning → French", size="sm") | |
| example_btn3 = gr.Button("🇩🇪 Thank you → German", size="sm") | |
| with gr.Row(): | |
| example_btn4 = gr.Button("🇯🇵 How are you? → Japanese", size="sm") | |
| example_btn5 = gr.Button("🇨🇳 Welcome → Chinese", size="sm") | |
| example_btn6 = gr.Button("🇰🇷 Goodbye → Korean", size="sm") | |
| with gr.Row(): | |
| example_btn7 = gr.Button("🇮🇳 Batch: Hello, Goodbye, Thank you → Hindi", size="sm") | |
| example_btn8 = gr.Button("🇷🇺 Business: Meeting, Project, Deadline → Russian", size="sm") | |
| gr.Markdown("") # Spacer | |
| translate_btn = gr.Button("Translate", variant="primary") | |
| # Middle Column: Supported Languages | |
| with gr.Column(scale=1): | |
| gr.Markdown("#### Supported Languages") | |
| gr.Markdown(""" | |
| **Common Language Codes:** | |
| - `es` - Spanish | |
| - `fr` - French | |
| - `de` - German | |
| - `hi` - Hindi | |
| - `ja` - Japanese | |
| - `ko` - Korean | |
| - `zh` - Chinese | |
| - `ar` - Arabic | |
| - `ru` - Russian | |
| - `pt` - Portuguese | |
| - `tr` - Turkish | |
| - `bn` - Bengali | |
| - `fi` - Finnish | |
| - `sw` - Swahili | |
| *For a complete list of 162+ supported languages, visit the [Language Codes Reference](https://jigsawstack.com/docs/additional-resources/languages)* | |
| """) | |
| # Example functions to auto-fill fields | |
| def fill_example_1(): | |
| return "Hello World", "es" | |
| def fill_example_2(): | |
| return "Good morning", "fr" | |
| def fill_example_3(): | |
| return "Thank you", "de" | |
| def fill_example_4(): | |
| return "How are you?", "ja" | |
| def fill_example_5(): | |
| return "Welcome", "zh" | |
| def fill_example_6(): | |
| return "Goodbye", "ko" | |
| def fill_example_7(): | |
| return "Hello, Goodbye, Thank you", "hi" | |
| def fill_example_8(): | |
| return "Meeting, Project, Deadline", "ru" | |
| # Connect example buttons to auto-fill functions | |
| example_btn1.click(fill_example_1, outputs=[text_input, target_lang]) | |
| example_btn2.click(fill_example_2, outputs=[text_input, target_lang]) | |
| example_btn3.click(fill_example_3, outputs=[text_input, target_lang]) | |
| example_btn4.click(fill_example_4, outputs=[text_input, target_lang]) | |
| example_btn5.click(fill_example_5, outputs=[text_input, target_lang]) | |
| example_btn6.click(fill_example_6, outputs=[text_input, target_lang]) | |
| example_btn7.click(fill_example_7, outputs=[text_input, target_lang]) | |
| example_btn8.click(fill_example_8, outputs=[text_input, target_lang]) | |
| gr.Markdown("#### Translation Result") | |
| translate_result = gr.JSON() | |
| translate_btn.click( | |
| translate_text, | |
| inputs=[text_input, target_lang], | |
| outputs=translate_result, | |
| ) | |
| demo.launch() | |