Update app.py

app.py

@@ -1,15 +1,21 @@
 import os
-os.system("playwright install")
 import re
 import urllib.parse
 import asyncio
 from typing import Dict, Optional
 from itertools import cycle
 
-import gradio as gr
+# Install playwright if not present
+os.system("playwright install")
+
+from flask import Flask, request, jsonify
 from bs4 import BeautifulSoup, NavigableString
 from playwright.async_api import async_playwright, TimeoutError as PlaywrightTimeoutError
 
+# --- Flask App Initialization ---
+app = Flask(__name__)
+
+# --- Credential and State Management (largely unchanged) ---
 class CredentialRevolver:
     def __init__(self, proxy_string: str):
         self.proxies = self._parse_proxies(proxy_string)
@@ -57,6 +63,7 @@ SEARCH_ENGINES = {
     "Perplexity": "https://www.perplexity.ai/search?q={query}",
 }
 
+# --- HTML to Markdown Conversion (unchanged) ---
 class HTML_TO_MARKDOWN_CONVERTER:
     def __init__(self, soup: BeautifulSoup, base_url: str):
         self.soup = soup
@@ -108,6 +115,7 @@ class HTML_TO_MARKDOWN_CONVERTER:
             return f"\n\n![{alt}]({full_src})\n\n"
         return inner_md
 
+# --- Core Web Browsing Logic (unchanged) ---
 async def perform_web_browse(action: str, query: str, browser_name: str, search_engine_name: str):
     browser_key = browser_name.lower()
     if "playwright" not in PLAYWRIGHT_STATE:
@@ -170,21 +178,54 @@ async def perform_web_browse(action: str, query: str, browser_name: str, search_
         if 'page' in locals() and not page.is_closed(): await page.close()
         if 'context' in locals(): await context.close()
 
-with gr.Blocks(title="Web Browse API", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Web Browse API")
-    gr.Markdown(f"This interface exposes a stateless API endpoint (`/api/web_browse`) to fetch and parse web content. {REVOLVER.count()} proxies loaded.")
-    
-    action_input = gr.Radio(label="Action", choices=["Search", "Scrape URL"], value="Search")
-    query_input = gr.Textbox(label="Query or URL", placeholder="e.g., 'best cat food' or 'www.wikipedia.org'")
-    
-    with gr.Row():
-        browser_input = gr.Dropdown(label="Browser", choices=["firefox", "chromium", "webkit"], value="firefox", scale=1)
-        search_engine_input = gr.Dropdown(label="Search Engine (if action is Search)", choices=sorted(list(SEARCH_ENGINES.keys())), value="DuckDuckGo", scale=2)
+
+# --- API Endpoint Definition ---
+@app.route('/web_browse', methods=['POST'])
+def web_browse():
+    """
+    API endpoint to perform a web search or scrape a URL.
+    This endpoint expects a JSON payload with the following parameters:
+    - "action": "Search" or "Scrape URL" (required)
+    - "query": The search term or the URL to scrape (required)
+    - "browser_name": "firefox", "chromium", or "webkit" (optional, default: "firefox")
+    - "search_engine_name": Name of the search engine (optional, default: "DuckDuckGo")
     
-    submit_button = gr.Button("Browse", variant="primary")
-    output_json = gr.JSON(label="API Result")
+    Example usage with curl:
+    curl -X POST http://127.0.0.1:5000/web_browse \
+         -H "Content-Type: application/json" \
+         -d '{
+               "action": "Search",
+               "query": "latest news on AI",
+               "browser_name": "firefox",
+               "search_engine_name": "Google"
+             }'
+    """
+    if not request.is_json:
+        return jsonify({"status": "error", "error_message": "Invalid input: payload must be JSON"}), 400
+
+    data = request.get_json()
+    action = data.get('action')
+    query = data.get('query')
+    browser_name = data.get('browser_name', 'firefox') # Default to firefox
+    search_engine_name = data.get('search_engine_name', 'DuckDuckGo') # Default to DuckDuckGo
+
+    if not action or not query:
+        return jsonify({"status": "error", "error_message": "Missing required parameters: 'action' and 'query' are mandatory."}), 400
     
-    submit_button.click(fn=perform_web_browse, inputs=[action_input, query_input, browser_input, search_engine_input], outputs=output_json, api_name="web_browse")
+    if action not in ["Search", "Scrape URL"]:
+        return jsonify({"status": "error", "error_message": "Invalid 'action'. Must be 'Search' or 'Scrape URL'."}), 400
+
+    # Run the async function in the current event loop
+    try:
+        result = asyncio.run(perform_web_browse(action, query, browser_name, search_engine_name))
+        status_code = 200 if result.get("status") == "success" else 500
+        return jsonify(result), status_code
+    except Exception as e:
+        return jsonify({"status": "error", "query": query, "error_message": f"An unexpected server error occurred: {str(e)}"}), 500
+
 
+# --- Main Application Runner ---
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    print(f"Flask server starting... {REVOLVER.count()} proxies loaded.")
+    print("API Endpoint available at POST http://127.0.0.1:7860/web_browse")
+    app.run(host='0.0.0.0', port=7860, debug=True)