search-api-webui 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

backend/app.py

@@ -1,30 +1,45 @@
 import json
 import os
+from pathlib import Path
 from flask import Flask, request, jsonify, send_from_directory
 from flask_cors import CORS
 from backend.providers import load_providers
 
+CURRENT_DIR = Path(__file__).resolve().parent
+STATIC_FOLDER = CURRENT_DIR / 'static'
+
 app = Flask(__name__, static_folder='static')
 CORS(app)
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-PROVIDERS_YAML = os.path.join(BASE_DIR, 'providers.yaml')
-USER_CONFIG_JSON = os.path.join(BASE_DIR, 'user_config.json')
+PROVIDERS_YAML = CURRENT_DIR / 'providers.yaml'
+USER_CONFIG_DIR = Path.home() / '.search-api-webui'
+USER_CONFIG_JSON = USER_CONFIG_DIR / 'config.json'
+
+if not USER_CONFIG_DIR.exists():
+    USER_CONFIG_DIR.mkdir(parents=True, exist_ok=True)
 
-provider_map = load_providers(PROVIDERS_YAML)
+if PROVIDERS_YAML.exists():
+    provider_map = load_providers(str(PROVIDERS_YAML))
+else:
+    print(f"Error: Configuration file not found at {PROVIDERS_YAML}")
+    provider_map = {}
 
 def get_stored_config():
-    if not os.path.exists(USER_CONFIG_JSON):
+    if not USER_CONFIG_JSON.exists():
         return {}
     try:
-        with open(USER_CONFIG_JSON, 'r') as f:
+        with open(USER_CONFIG_JSON, 'r', encoding='utf-8') as f:
             return json.load(f)
-    except:
+    except Exception as e:
+        print(f"Error reading config: {e}")
         return {}
 
 def save_stored_config(config_dict):
-    with open(USER_CONFIG_JSON, 'w') as f:
-        json.dump(config_dict, f, indent=2)
+    try:
+        with open(USER_CONFIG_JSON, 'w', encoding='utf-8') as f:
+            json.dump(config_dict, f, indent=2)
+    except Exception as e:
+        print(f"Error saving config: {e}")
 
 @app.route('/api/providers', methods=['GET'])
 def get_providers_list():
@@ -127,10 +142,10 @@ def search_api():
 @app.route('/', defaults={'path': ''})
 @app.route('/<path:path>')
 def serve(path):
-    if path != "" and os.path.exists(os.path.join(app.static_folder, path)):
-        return send_from_directory(app.static_folder, path)
+    if path != "" and (STATIC_FOLDER / path).exists():
+        return send_from_directory(str(STATIC_FOLDER), path)
     else:
-        return send_from_directory(app.static_folder, 'index.html')
+        return send_from_directory(str(STATIC_FOLDER), 'index.html')
 
 def main():
     import argparse
@@ -138,7 +153,11 @@ def main():
     parser.add_argument("--port", type=int, default=8889, help="Port to run the server on")
     parser.add_argument("--host", type=str, default="0.0.0.0", help="Host to run the server on")
     args = parser.parse_args()
-    
+
+    print(f"Starting Search API WebUI...")
+    print(f"  - Config Storage: {USER_CONFIG_JSON}")
+    print(f"  - Serving on: http://{args.host}:{args.port}")
+
     app.run(host=args.host, port=args.port)
 
 if __name__ == "__main__":

backend/providers/__init__.py

@@ -26,7 +26,7 @@ def load_providers(file_path='providers.yaml'):
         provider_type = conf.get('type', 'generic')
 
         # Instantiate specific provider based on type or name
-        if name == 'querit' or provider_type == 'querit_sdk':
+        if provider_type == 'querit_sdk':
             providers[name] = QueritSdkProvider(conf)
         else:
             providers[name] = GenericProvider(conf)

backend/providers/generic.py

@@ -17,6 +17,9 @@ class GenericProvider(BaseProvider):
             config (dict): Configuration containing url, headers, params, and mapping rules.
         """
         self.config = config
+        self.session = requests.Session()  # Persistent connection session
+        self._connection_ready = False     # Connection ready status
+        self._last_url = None              # Last used URL tracker
 
     def _fill_template(self, template_obj, **kwargs):
         """
@@ -42,6 +45,31 @@ class GenericProvider(BaseProvider):
             return {k: self._fill_template(v, **kwargs) for k, v in template_obj.items()}
         return template_obj
 
+    def _ensure_connection(self, url, headers):
+        """
+        Pre-warm HTTPS connection and verify availability.
+        Uses lightweight HEAD request to verify connection without fetching response body.
+
+        Args:
+            url (str): Target URL
+            headers (dict): Request headers
+
+        Returns:
+            bool: Whether connection is ready
+        """
+        # Re-warm if URL changed or connection not ready
+        if url != self._last_url or not self._connection_ready:
+            try:
+                # Verify connection using HEAD request (no response body)
+                self.session.head(url, headers=headers, timeout=5)
+                self._connection_ready = True
+                self._last_url = url
+                print(f'  [Connection Pool] Connected to: {url}')
+            except Exception as e:
+                self._connection_ready = False
+                print(f'  [Connection Pool] Connection warm-up failed: {e}')
+                raise
+
     def search(self, query, api_key, **kwargs):
         # 1. Extract parameters with defaults
         limit = kwargs.get('limit', '10')
@@ -69,6 +97,18 @@ class GenericProvider(BaseProvider):
         print(f'[{self.config.get("name", "Unknown")}] Search:')
         print(f'  URL: {url} | Method: {method}')
         
+        # Ensure connection is pre-warmed (use HEAD request to verify availability)
+        # Pre-warming is not counted in request latency, only verifies connection
+        try:
+            self._ensure_connection(url, headers)
+        except Exception as e:
+            print(f"Connection Warm-up Error: {e}")
+            return {
+                "error": f"Connection failed: {str(e)}",
+                "results": [],
+                "metrics": {"latency_ms": 0, "size_bytes": 0}
+            }
+
         start_time = time.time()
 
         try:
@@ -78,10 +118,11 @@ class GenericProvider(BaseProvider):
             if json_body:
                 req_args['json'] = json_body
 
+            # Use Session to send request (connection is reused)
             if method.upper() == 'GET':
-                response = requests.get(url, **req_args)
+                response = self.session.get(url, **req_args)
             else:
-                response = requests.post(url, **req_args)
+                response = self.session.post(url, **req_args)
 
             response.raise_for_status()
         except Exception as e:

backend/providers.yaml

@@ -1,7 +1,18 @@
 querit:
-  type: "querit_sdk"
-  description: "Official Querit Search via Python SDK"
-  default_limit: 10
+  url: "https://api.querit.ai/v1/search"
+  method: "POST"
+  headers:
+    "Accept": "application/json"
+    "Authorization": "Bearer {api_key}"
+    "Content-Type": "application/json"
+  payload:
+    query: "{query}"
+  response_mapping:
+    root_path: "results.result"
+    fields:
+      title: "title"
+      url: "url"
+      snippet: "snippet"
 
 ydc_search:
   url: "https://ydc-index.io/v1/search"
@@ -17,3 +28,8 @@ ydc_search:
       title: "title"
       url: "url"
       snippet: "snippets[0] || description"
+
+querit_sdk:
+  type: "querit_sdk"
+  description: "Official Querit Search via Python SDK"
+  default_limit: 10

search_api_webui-0.1.2.dist-info/METADATA

@@ -0,0 +1,122 @@
+Metadata-Version: 2.4
+Name: search-api-webui
+Version: 0.1.2
+Summary: A Search API WebUI for Querit, You, and other search providers.
+Project-URL: Homepage, https://github.com/querit-ai/search-api-webui
+Project-URL: Repository, https://github.com/querit-ai/search-api-webui.git
+Project-URL: Issues, https://github.com/querit-ai/search-api-webui/issues
+Author: querit.ai
+License: MIT
+License-File: LICENSE
+Keywords: api,llm,querit,search,tool,webui
+Classifier: Development Status :: 3 - Alpha
+Classifier: Framework :: Flask
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.7
+Requires-Dist: flask-cors
+Requires-Dist: flask>=2.0.0
+Requires-Dist: jmespath
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: querit
+Requires-Dist: requests>=2.25.0
+Description-Content-Type: text/markdown
+
+# Search API WebUI
+
+A lightweight, local WebUI for testing and visualizing Search APIs (Querit, YDC, etc.).
+
+(images)
+
+## Features
+
+**Search**: Support for Querit, You.com, and generic Search APIs via configuration.
+**Performance Metrics**: Real-time display of request latency and payload size.
+**Visual Rendering**: Renders standard search results (Title, Snippet, URL) in a clean card layout.
+**Configurable**: Easy-to-edit providers.yaml to add or modify search providers.
+**Secure**: API Keys are stored locally in user_config.json and never committed.
+## Installation
+
+Use this method if you just want to run the tool without modifying the code.
+
+### Prerequisites
+
+Python 3.7+
+### Install via Pip
+
+```
+pip install search-api-webui
+```
+
+### Run the Server
+
+```
+search-api-webui
+```
+
+Open your browser at http://localhost:8889.
+
+## Development
+
+Use this method if you want to contribute to the code or build from source.
+
+### Prerequisites
+
+Python 3.7+
+Node.js & npm (for building the frontend)
+### Setup Steps
+
+**Clone the repository**
+```
+git clone https://github.com/querit-ai/search-api-webui.git
+cd search-api-webui
+```
+
+**Build Frontend**
+```
+cd frontend
+npm install
+npm run build
+cd …
+```
+
+**Install Backend (Editable Mode)**
+```
+pip install -e .
+```
+
+**Run the Server**
+```
+python -m backend.app
+```
+
+## Configuration
+
+### Add API Keys
+
+Open the WebUI settings page (click the gear icon).
+Enter your API Key for the selected provider (e.g., Querit).
+Keys are saved locally in user_config.json.
+### Add New Providers
+
+Edit providers.yaml in the root directory to add custom API endpoints. The system uses JMESPath to map JSON responses to the UI.
+
+```
+my_custom_search:
+url: “https://api.example.com/search”
+method: “GET”
+headers:
+Authorization: “Bearer {api_key}”
+params:
+q: “{query}”
+response_mapping:
+root_path: “data.items”
+fields:
+title: “title”
+url: “link”
+snippet: “snippet”
+```
+
+## License
+
+MIT License. See LICENSE for details.

search_api_webui-0.1.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+backend/app.py,sha256=XLW8bJVgap4yZ6MUgAuJ4bxSID_zKM699nCW7L_e3XA,5144
+backend/providers/__init__.py,sha256=1RfIA8eQxDLDjXrUxWzmg3tucZGv7WuZ5pJBsc6fAnQ,1052
+backend/providers/base.py,sha256=_iYcxCxI7VCoxb02D71KqtojAU91D6rc3c5dwKNYJ2o,863
+backend/providers/generic.py,sha256=ztwqko1KjiC2BS2afwZY2heREWLW0Kh2ZivqgLofbXU,6211
+backend/providers/querit.py,sha256=5_vHXcDz59jhqY35w8vaoVH3BqGiQCYEZaekIvTKkak,2688
+backend/providers.yaml,sha256=GMF7-o9eqe1AVXuMHEfdfl4OPeIZInDCqjsvO2tRHp0,752
+backend/static/index.html,sha256=fhLWDY2W-bN-bBq2fmkch-e6gRZpoyqbKvNBzpWHiHk,401
+backend/static/assets/index-CF13bI2g.js,sha256=91wc8meWvRXloy4vzmGN43DvrpHIjMVIdCihXCw9zBU,208926
+backend/static/assets/index-DLyBd1PD.css,sha256=nevUEbN7Fw9cq3aXWV7v_tQpD00d7kr3I64dgZ-e7ys,17914
+search_api_webui-0.1.2.dist-info/METADATA,sha256=0_qkPbasFwqzvM-53fSlgwyAvo8cPwSYZ-nPq3007I0,2827
+search_api_webui-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+search_api_webui-0.1.2.dist-info/entry_points.txt,sha256=D9R5N6s2Bt8rEKYSINTUB1f45nH-Xc5iIKNPfYYc8ec,54
+search_api_webui-0.1.2.dist-info/licenses/LICENSE,sha256=jHhlLwtvhZRz1yU1G_3cjsPigCTXQGT1qshhyekdmTE,1061
+search_api_webui-0.1.2.dist-info/RECORD,,

search_api_webui-0.1.0.dist-info/METADATA

@@ -1,93 +0,0 @@
-Metadata-Version: 2.4
-Name: search-api-webui
-Version: 0.1.0
-Summary: A Search API WebUI for Querit, You, and other search providers.
-Project-URL: Homepage, https://github.com/querit-ai/search-api-webui
-Project-URL: Repository, https://github.com/querit-ai/search-api-webui.git
-Project-URL: Issues, https://github.com/querit-ai/search-api-webui/issues
-Author: Search API WebUI Team
-License: MIT
-License-File: LICENSE
-Keywords: api,llm,querit,search,tool,webui
-Classifier: Development Status :: 3 - Alpha
-Classifier: Framework :: Flask
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.7
-Requires-Dist: flask-cors
-Requires-Dist: flask>=2.0.0
-Requires-Dist: jmespath
-Requires-Dist: pyyaml>=6.0
-Requires-Dist: querit
-Requires-Dist: requests>=2.25.0
-Description-Content-Type: text/markdown
-
-# Search API WebUI
-
-A lightweight, local WebUI for testing and visualizing Search APIs (Querit, YDC, etc.).
-
-## Features
-
-- 🔍 **Search**: Support for Querit, You.com, and generic Search APIs via configuration.
-- ⚡ **Performance Metrics**: Real-time display of request latency and payload size.
-- 🎨 **Visual Rendering**: Renders standard search results (Title, Snippet, URL) in a clean card layout.
-- 🛠️ **Configurable**: Easy-to-edit `providers.yaml` to add or modify search providers.
-- 🔒 **Secure**: API Keys are stored locally in `user_config.json` and never committed.
-
-## Quick Start
-
-### Prerequisites
-
-- Python 3.7+
-- Node.js & npm (for building the frontend)
-
-### Installation
-
-1. **Clone the repository**
-   ```bash
-   git clone https://github.com/querit-ai/search-api-webui.git
-   cd search-api-webui
-   ```
-2. **Build Frontend**
-   ```bash
-   cd frontend
-   npm install
-   npm run build
-   cd ..
-   ```
-3. **Install Backend**
-   ```bash
-   pip install -e .
-   ```
-
-### Usage
-**Run the server:**
-   ```bash
-   python -m backend.app
-   ```
-Or if you installed via pip:
-   ```bash
-   search-api-webui
-   ```
-
-Open your browser at http://localhost:8889.
-
-## Configuration
-### Add API Keys
-1. Open the WebUI settings page.
-2. Enter your API Key for the selected provider (e.g., Querit).
-3. Keys are saved locally in user_config.json.
-
-### Add New Providers
-Edit providers.yaml to add custom API endpoints. The system uses JMESPath to map JSON responses to the UI.
-```yaml
-my_custom_search:
-      url: "https://api.example.com/search"
-      method: "GET"
-      ...
-```
-## Development
-Backend: Flask (in backend/)
-Frontend: React + Vite (in frontend/)
-## License
-MIT License. See LICENSE for details.

search_api_webui-0.1.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-backend/app.py,sha256=J5GB0-rNOPq5Ny0DK5LDO_v0OqvUiMF7MM5Xk29aviM,4540
-backend/providers/__init__.py,sha256=l9SCd9sEISWZB-E2Co8ds2hILaxBVoh2S6SSg7MXXIo,1072
-backend/providers/base.py,sha256=_iYcxCxI7VCoxb02D71KqtojAU91D6rc3c5dwKNYJ2o,863
-backend/providers/generic.py,sha256=3dyx3Hpbo4hlfUcwy1wYt5N5dc2X3fK1iFnOoT3m_Vc,4440
-backend/providers/querit.py,sha256=5_vHXcDz59jhqY35w8vaoVH3BqGiQCYEZaekIvTKkak,2688
-backend/providers.yaml,sha256=2aRmfT7NmeU9IfUqX7phRjOf_fUACRD5ondYJO2HH4I,400
-backend/static/index.html,sha256=fhLWDY2W-bN-bBq2fmkch-e6gRZpoyqbKvNBzpWHiHk,401
-backend/static/assets/index-CF13bI2g.js,sha256=91wc8meWvRXloy4vzmGN43DvrpHIjMVIdCihXCw9zBU,208926
-backend/static/assets/index-DLyBd1PD.css,sha256=nevUEbN7Fw9cq3aXWV7v_tQpD00d7kr3I64dgZ-e7ys,17914
-search_api_webui-0.1.0.dist-info/METADATA,sha256=1xdpdSNV2TPkAGfio11onBkZp8yPjEn6Axk4KiJDAcM,2577
-search_api_webui-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-search_api_webui-0.1.0.dist-info/entry_points.txt,sha256=D9R5N6s2Bt8rEKYSINTUB1f45nH-Xc5iIKNPfYYc8ec,54
-search_api_webui-0.1.0.dist-info/licenses/LICENSE,sha256=jHhlLwtvhZRz1yU1G_3cjsPigCTXQGT1qshhyekdmTE,1061
-search_api_webui-0.1.0.dist-info/RECORD,,