Upload app.py with huggingface_hub

app.py

@@ -1,10 +1,15 @@
 # -*- coding: utf-8 -*-
 import os
 import sqlite3
+import json
+import requests
 from flask import Flask, request, jsonify, g
 
 app = Flask(__name__)
 
+# ----------------------------------------------------------------------
+# Database utilities (unchanged)
+# ----------------------------------------------------------------------
 DATABASE = os.path.join(os.path.dirname(__file__), "tasks.db")
 
 def get_db():
@@ -35,6 +40,9 @@ def close_connection(exception):
     if db is not None:
         db.close()
 
+# ----------------------------------------------------------------------
+# Existing task endpoints (unchanged)
+# ----------------------------------------------------------------------
 @app.route("/tasks", methods=["GET"])
 def list_tasks():
     db = get_db()
@@ -85,6 +93,9 @@ def delete_task(task_id):
     db.commit()
     return "", 204
 
+# ----------------------------------------------------------------------
+# Health endpoints (unchanged)
+# ----------------------------------------------------------------------
 @app.route("/", methods=["GET"])
 def index():
     return "Task API is running."
@@ -93,10 +104,76 @@ def index():
 def ping():
     return "pong", 200
 
+# ----------------------------------------------------------------------
+# New: Generic API connector
+# ----------------------------------------------------------------------
+@app.route("/api/connector", methods=["POST"])
+def api_connector():
+    """
+    Expects a JSON payload with the following fields:
+    {
+        "url": "https://example.com/endpoint",
+        "method": "GET|POST|PUT|DELETE|PATCH",
+        "headers": {"Authorization": "Bearer ...", ...},   # optional
+        "params": {"q": "search"},                         # optional, for query string
+        "json": {"key": "value"}                          # optional, for JSON body
+    }
+    The endpoint forwards the request to the target API and returns
+    the raw response (status code, headers and JSON body if possible).
+    """
+    payload = request.get_json()
+    if not payload:
+        return jsonify({"error": "invalid json payload"}), 400
+
+    url = payload.get("url")
+    method = payload.get("method", "GET").upper()
+    headers = payload.get("headers", {})
+    params = payload.get("params", {})
+    json_body = payload.get("json", None)
+
+    if not url:
+        return jsonify({"error": "url is required"}), 400
+
+    # Optional: inject a default API key from environment if not provided
+    default_key = os.environ.get("DEFAULT_API_KEY")
+    if default_key and "Authorization" not in headers:
+        headers["Authorization"] = f"Bearer {default_key}"
+
+    try:
+        response = requests.request(
+            method=method,
+            url=url,
+            headers=headers,
+            params=params,
+            json=json_body,
+            timeout=30
+        )
+    except requests.RequestException as e:
+        return jsonify({"error": "request failed", "details": str(e)}), 502
+
+    # Try to parse JSON, fallback to raw text
+    try:
+        resp_content = response.json()
+    except ValueError:
+        resp_content = response.text
+
+    result = {
+        "status_code": response.status_code,
+        "headers": dict(response.headers),
+        "content": resp_content
+    }
+    return jsonify(result), response.status_code
+
+# ----------------------------------------------------------------------
+# Error handling (unchanged)
+# ----------------------------------------------------------------------
 @app.errorhandler(404)
 def not_found(e):
     return jsonify({"error": "not found"}), 404
 
+# ----------------------------------------------------------------------
+# Main entry point (unchanged)
+# ----------------------------------------------------------------------
 if __name__ == "__main__":
     if not os.path.isfile(DATABASE):
         init_db()