import gradio as gr import requests import json import os import time from collections import defaultdict 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 Wrasppers ------------------ def summarize(text, url, file_store_key, summary_type, max_points, max_characters, request: gr.Request): """Summarize text, URL, or file store key using JigsawStack Summary API""" # Check rate limit first rate_limit_ok, rate_limit_msg = check_rate_limit(request) if not rate_limit_ok: return rate_limit_msg # Validate that at least one input method is provided if not text and not url and not file_store_key: return "Error: Please provide either text, URL, or file store key" payload = {} # Add text if provided if text and text.strip(): payload["text"] = text.strip() # Add URL if provided if url and url.strip(): payload["url"] = url.strip() # Add file store key if provided if file_store_key and file_store_key.strip(): payload["file_store_key"] = file_store_key.strip() # Add summary type if summary_type: payload["type"] = summary_type # Add max points if provided and type is points if summary_type == "points" and max_points: try: max_points_int = int(max_points) if 1 <= max_points_int <= 100: payload["max_points"] = max_points_int else: return "Error: max_points must be between 1 and 100" except ValueError: return "Error: max_points must be a valid number" # Add max characters if provided if max_characters: try: max_chars_int = int(max_characters) if max_chars_int > 0: payload["max_characters"] = max_chars_int else: return "Error: max_characters must be greater than 0" except ValueError: return "Error: max_characters must be a valid number" try: r = requests.post(f"{BASE_URL}/ai/summary", headers=headers, json=payload) r.raise_for_status() result = r.json() if not result.get("success"): return f"Error: {result.get('message', 'Unknown error')}" summary = result.get("summary", "") # Format the output based on type if summary_type == "points" and isinstance(summary, list): return "\n• " + "\n• ".join(summary) else: return summary except requests.exceptions.RequestException as req_err: return f"Request failed: {str(req_err)}" except Exception as e: return f"Unexpected error: {str(e)}" # ----------------- Gradio UI ------------------ with gr.Blocks() as demo: gr.Markdown("""

🧩 Text Summarizer

Generate concise, intelligent summaries of text, URLs, or stored documents with multiple output formats.

For more details and API usage, see the documentation.

""") with gr.Row(): with gr.Column(): gr.Markdown("#### Input Method (select one)") input_method = gr.Radio( choices=["Text", "URL", "File Store Key"], label="Choose Input Method", value="Text" ) # Conditional inputs based on selection long_text = gr.Textbox(label="Text to Summarize", lines=8, placeholder="Enter your text here...") url_input = gr.Textbox(label="Document URL (PDF)", placeholder="https://example.com/document.pdf", visible=False) file_key = gr.Textbox(label="File Store Key", placeholder="your-file-store-key", visible=False) with gr.Column(): gr.Markdown("#### Summary Options") summary_type = gr.Radio( choices=["text", "points"], label="Summary Format", value="text", info="Text: continuous paragraph | Points: bullet points" ) max_points = gr.Slider( label="Max Points (2-100)", value=5, minimum=2, maximum=100, step=1, info="Only applies when format is 'points'" ) max_characters = gr.Number( label="Max Characters", value=200, minimum=1, info="Optional: limit summary length" ) summary_btn = gr.Button("Generate Summary", variant="primary") summary = gr.Textbox(label="Generated Summary", lines=10) # Function to show/hide input groups based on selection def update_input_visibility(method): if method == "Text": return gr.Textbox(visible=True), gr.Textbox(visible=False), gr.Textbox(visible=False) elif method == "URL": return gr.Textbox(visible=False), gr.Textbox(visible=True), gr.Textbox(visible=False) elif method == "File Store Key": return gr.Textbox(visible=False), gr.Textbox(visible=False), gr.Textbox(visible=True) else: return gr.Textbox(visible=True), gr.Textbox(visible=False), gr.Textbox(visible=False) input_method.change( update_input_visibility, inputs=input_method, outputs=[long_text, url_input, file_key] ) summary_btn.click( summarize, inputs=[long_text, url_input, file_key, summary_type, max_points, max_characters], outputs=summary ) demo.launch()