refactor: refactor code for gradio client

README.md

@@ -10,32 +10,7 @@ pinned: false
 
 # Proxy Service
 
-🔒 **Сервисное приложение для проброса вебхуков Яндекс.Форм**
-
-## ⚠️ ВНИМАНИЕ
-
-Это **сервисное приложение**, предназначенное исключительно для системной интеграции.  
-
-## 📋 Назначение
-
-Этот прокси-сервер перенаправляет входящие вебхуки в приватное пространство.
-
-## 📊 Доступные эндпоинты
-
-| Метод | Путь | Назначение |
-|-------|------|------------|
-| `GET`  | `/ui` | **Интерфейс администратора** (Проверка статуса) |
-| `POST` | `/{SECRET_PATH}` | **Прием вебхуков** (Только известный путь) |
-| `GET`  | `/health` | Проверка работоспособности (Health Check) |
-| `GET`  | `/api` | Техническая информация о сервисе |
-
-> **Примечание:** Вебхуки принимаются только на путь, указанный в переменной окружения `PUBLIC_ENDPOINT`. Все остальные запросы возвращают 404.
-
-## 🛠 Конфигурация
-
-Для работы требуются Secrets:
-- `PRIVATE_SPACE_URL`: Полный URL целевого сервиса
-- `PUBLIC_ENDPOINT`: Секретный путь для приема вебхуков
+🔒 **Сервисное приложение для проброса вебхуков**
 
 ## 📞 Поддержка
 

app.py

@@ -1,35 +1,44 @@
 import os
-import requests
+import json
 import gradio as gr
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse, RedirectResponse
+from gradio_client import Client
 
-# Конфигурация
-PRIVATE_SPACE_URL = os.getenv("PRIVATE_SPACE_URL")
+# ==================== Конфигурация ====================
+PRIVATE_SPACE_ID = os.getenv("PRIVATE_SPACE_ID")
+PRIVATE_TOKEN = os.getenv("PRIVATE_TOKEN")
 PUBLIC_ENDPOINT = os.getenv("PUBLIC_ENDPOINT")
+PRIVATE_ENDPOINT = os.getenv("PRIVATE_ENDPOINT")
 
-# ==================== FastAPI приложение ====================
-fastapi_app = FastAPI(title="Proxy Service")
+# ==================== Инициализация Клиента ====================
+client = None
 
 
-# Health check
-@fastapi_app.get("/health")
-async def health_check():
-    """Безопасная проверка работоспособности."""
-    is_configured = bool(PRIVATE_SPACE_URL and PUBLIC_ENDPOINT)
-    return JSONResponse(
-        content={
-            "status": "healthy" if is_configured else "degraded",
-            "service": "proxy",
-            "timestamp": "...",
-        }
-    )
+def init_client():
+    """Ленивая или повторная инициализация клиента"""
+    global client
+    if not PRIVATE_SPACE_ID or not PRIVATE_TOKEN:
+        print("⚠️  Config missing: PRIVATE_SPACE_ID or HF_TOKEN not set.")
+        return None
+
+    print("🔄 Initializing upstream connection...")
+    try:
+        # Клиент создается
+        new_client = Client(PRIVATE_SPACE_ID, hf_token=PRIVATE_TOKEN, verbose=False)
+        print("✅ Upstream connection established.")
+        return new_client
+    except Exception as e:
+        # Логируем только факт ошибки, не печатаем само исключение, так как оно может содержать URL
+        print("❌ Connection failed.")
+        return None
+
 
+# Пробуем соединиться при старте
+client = init_client()
 
-# Корневой эндпоинт
-@fastapi_app.get("/api")
-async def root():
-    return JSONResponse(content={"service": "API Gateway", "status": "operational"})
+# ==================== FastAPI (Прокси) ====================
+fastapi_app = FastAPI(title="Secure Proxy")
 
 
 @fastapi_app.get("/")
@@ -37,100 +46,75 @@ async def root_redirect():
     return RedirectResponse(url="/ui")
 
 
-# ==================== Безопасный Gradio интерфейс ====================
-def create_gradio_interface():
-    with gr.Blocks(title="Proxy Service") as demo:
-        gr.Markdown("# 🔒 Защищенный прокси-сервис")
-        gr.Markdown(
-            """
-            ### ⚠️ Сервисное приложение
-            Этот сервис предназначен исключительно для системной интеграции.
-            
-            ---
-            
-            ### 📊 Системный статус
-            """
-        )
-
-        # Безопасный статус
-        status_box = gr.Textbox(label="Статус", interactive=False, lines=4)
-
-        def get_safe_status():
-            """Возвращает безопасный статус без конфиденциальной информации."""
-            if not PUBLIC_ENDPOINT or not PRIVATE_SPACE_URL:
-                return "🔧 Требуется настройка\n\nУстановите необходимые переменные окружения."
-
-            return """✅ Сервис готов к работе
+@fastapi_app.get("/health")
+async def health_check():
+    # Не показываем детали, просто жив сервис или нет
+    return {"status": "ok" if client else "connecting"}
 
-Статус: Активен
-Вебхук: Настроен
-Целевой сервис: Доступен
 
-Для настройки интеграции обратитесь к администратору системы."""
+@fastapi_app.post(f"/{PUBLIC_ENDPOINT}")
+async def proxy_webhook(request: Request):
+    global client
 
-        # Связываем кнопку со статусом
-        refresh_btn = gr.Button("Обновить статус")
-        refresh_btn.click(get_safe_status, outputs=status_box)
+    # Если клиент отвалился или не был создан, пробуем пересоздать
+    if not client:
+        client = init_client()
+        if not client:
+            raise HTTPException(status_code=503, detail="Upstream unavailable")
 
-        # Загружаем статус при загрузке
-        demo.load(get_safe_status, outputs=status_box)
+    try:
+        # 1. Получаем JSON (без логирования содержимого)
+        payload = await request.json()
+
+        # 2. Конвертируем в строку для Gradio
+        payload_str = json.dumps(payload, ensure_ascii=False)
+
+        # 3. Вызываем приватный API
+        # Используем api_name, это безопаснее при изменениях кода.
+        result = client.predict(payload_str, api_name=f"/{PRIVATE_ENDPOINT}")
+
+        # 4. Возвращаем успех, не показывая ответ приватного сервера (если там есть данные)
+        # Если нужно вернуть ID для отладки Яндекс.Форм, можно вернуть минимальный JSON
+        return JSONResponse(
+            content={
+                "status": "forwarded",
+                "timestamp": os.urandom(4).hex(),  # Просто уникальный маркер
+            }
+        )
 
-    return demo
+    except Exception:
+        # В случае ошибки не возвращаем её текст клиенту, чтобы не светить внутренности
+        print("❌ Error during forwarding.")
+        raise HTTPException(status_code=500, detail="Internal forwarding error")
 
 
-# Создаем интерфейс
-demo = create_gradio_interface()
+# ==================== Безопасный Gradio UI ====================
+def create_secure_ui():
+    with gr.Blocks(title="System Status") as demo:
+        gr.Markdown("## 🛡️ Gateway Status")
 
-# Принудительно прописываем путь корня (нужно для корректной работы JS/CSS)
-demo.root_path = "/ui"
+        # Индикатор статуса без деталей
+        status_indicator = gr.Textbox(
+            label="Service State", value="Checking...", interactive=False
+        )
 
-# КЛЮЧЕВОЕ ИЗМЕНЕНИЕ: монтируем Gradio на /ui, а не на /
-app = gr.mount_gradio_app(fastapi_app, demo, path="/ui", root_path="/ui")
+        btn = gr.Button("Refresh Status")
 
+        def check_safe():
+            if client:
+                return "✅ Online / Connected to Upstream"
+            else:
+                return "⚠️ Offline / Reconnecting..."
 
-# Универсальный обработчик
-@fastapi_app.post("/{path:path}")
-async def proxy_webhook(path: str, request: Request):
-    """Обработчик вебхуков с проверкой пути."""
-    # Проверяем секретный путь
-    if not PUBLIC_ENDPOINT or path != PUBLIC_ENDPOINT:
-        raise HTTPException(status_code=404, detail="Not found")
+        # При загрузке и нажатии проверяем статус
+        demo.load(check_safe, outputs=status_indicator)
+        btn.click(check_safe, outputs=status_indicator)
 
-    if not PRIVATE_SPACE_URL:
-        raise HTTPException(status_code=500, detail="Service error")
+    return demo
 
-    try:
-        body_bytes = await request.body()
-
-        # Минимальные заголовки
-        headers = {
-            "Content-Type": request.headers.get("content-type", "application/json"),
-        }
-
-        # Только заголовки
-        incoming_headers = ["X-FORM-ID", "X-DELIVERY-ID", "X-FORM-ANSWER-ID"]
-        for header in incoming_headers:
-            if value := request.headers.get(header):
-                headers[header] = value
-
-        # Отправляем запрос
-        response = requests.post(
-            PRIVATE_SPACE_URL, data=body_bytes, headers=headers, timeout=25
-        )
-        print("data:", body_bytes)
-        print("headers:", headers)
-        print("PRIVATE_SPACE_URL:", PRIVATE_SPACE_URL)
-        print("JSONResponse:", JSONResponse(response.json()))
-        # Возвращаем ответ
-        return JSONResponse(response.json())
-
-    except requests.exceptions.Timeout:
-        raise HTTPException(status_code=504, detail="Service timeout")
-    except Exception:
-        raise HTTPException(status_code=500, detail="Internal error")
 
+app = gr.mount_gradio_app(fastapi_app, create_secure_ui(), path="/ui")
 
-# Локальный запуск
 if __name__ == "__main__":
     import uvicorn
 

requirements.txt

@@ -1,4 +1,5 @@
-gradio
 fastapi==0.104.1
 uvicorn[standard]==0.24.0
-requests==2.31.0
+python-multipart==0.0.6
+gradio>=4.0.0
+gradio_client>=0.8.0