Update app.py

app.py

@@ -134,15 +134,50 @@ def enhanced_ai_scrape(input_method, url, html, prompts_str, selector, page_pos,
     except Exception as e:
         return error_response(f"Unexpected error: {str(e)}")
 
+
+def get_rate_limit_status(request: gr.Request):
+    """Get current rate limit status for the user"""
+    if not request:
+        return {"error": "Unable to get request info"}
+    
+    ip = get_real_ip(request)
+    now = time.time()
+    
+    # Clean up old timestamps
+    request_times[ip] = [t for t in request_times[ip] if now - t < TIME_WINDOW]
+    
+    current_requests = len(request_times[ip])
+    time_window_minutes = round(TIME_WINDOW / 60, 1)
+    
+    if current_requests >= MAX_REQUESTS:
+        time_remaining = int(TIME_WINDOW - (now - request_times[ip][0]))
+        time_remaining_minutes = round(time_remaining / 60, 1)
+        return {
+            "status": "Rate limited",
+            "current_requests": current_requests,
+            "max_requests": MAX_REQUESTS,
+            "time_window_minutes": time_window_minutes,
+            "time_remaining_minutes": time_remaining_minutes
+        }
+    else:
+        return {
+            "status": "Available",
+            "current_requests": current_requests,
+            "max_requests": MAX_REQUESTS,
+            "time_window_minutes": time_window_minutes,
+            "remaining_requests": MAX_REQUESTS - current_requests
+        }
+
 # ----------------- 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;'>🧩 AI Scraper</h1>
-        <p style='font-size:1.2em; margin-top: 0;'>Extract structured data from web pages with advanced AI models.</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/scrape' target='_blank'>documentation</a>.</p>
-    </div>
+        <div style='text-align: center; margin-bottom: 24px;'>
+            <h1 style='font-size:2.2em; margin-bottom: 0.2em;'>🧩 AI Scraper</h1>
+            <p style='font-size:1.2em; margin-top: 0;'>Extract structured data from web pages with advanced AI models.</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/scrape' target='_blank'>documentation</a>.</p>
+            <p style='font-size:0.9em; margin-top: 0.5em; color: #666;'>Rate limit: 1 request per hour per IP address</p>
+        </div>
     """)
 
     with gr.Row():
@@ -225,7 +260,7 @@ with gr.Blocks() as demo:
 
             # Rate limit status
             gr.Markdown("#### Rate Limit Status")
-            rate_limit_status = gr.JSON(label="Current Usage", interactive=False)
+            rate_limit_status = gr.JSON(label="Current Usage")
             refresh_status_btn = gr.Button("Refresh Status", size="sm")
 
     scrape_btn = gr.Button("Scrape with AI", variant="primary")
@@ -248,7 +283,24 @@ with gr.Blocks() as demo:
     scrape_btn.click(
         enhanced_ai_scrape,
         inputs=[input_method_scraper, url_scraper, html_content, element_prompts, root_selector, page_position],
-        outputs=[scrape_status, context_output, selectors_output, detailed_data, links_data, pagination_info]
+        outputs=[scrape_status, context_output, selectors_output, detailed_data, links_data, pagination_info],
+        _js="() => []"
+    )
+    
+    # Rate limit status handlers
+    refresh_status_btn.click(
+        get_rate_limit_status,
+        inputs=[],
+        outputs=rate_limit_status,
+        _js="() => []"
+    )
+    
+    # Auto-refresh rate limit status when page loads
+    demo.load(
+        get_rate_limit_status,
+        inputs=[],
+        outputs=rate_limit_status,
+        _js="() => []"
     )
 
 demo.launch()