import gradio as gr import requests from PIL import Image import io import os BASE_URL = "https://api.jigsawstack.com/v1" headers = {"x-api-key": os.getenv("JIGSAWSTACK_API_KEY")} # ----------------- JigsawStack API Wrasppers ------------------ def summarize(text, url, file_store_key, summary_type, max_points, max_characters): # 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: print(payload) 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()