Upload 2 files

app.py

@@ -0,0 +1,148 @@
+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("""
+    <div style='text-align: center; margin-bottom: 24px;'>
+        <h1 style='font-size:2.2em; margin-bottom: 0.2em;'>🧩 JigsawStack Image Translation</h1>
+        <p style='font-size:1.2em; margin-top: 0;'>Extract and translate text from images into multiple languages.</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/image-translate' target='_blank'>documentation</a>.</p>
+    </div>
+    """)
+
+    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 (1-100)",
+                value=5,
+                minimum=1,
+                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]
+    )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+requests
+Pillow