Create proxy.py

proxy.py

@@ -0,0 +1,34 @@
+import os
+from flask import Flask, request, Response
+import requests
+
+app = Flask(__name__)
+
+# Pega a chave que você vai definir nas configurações do Hugging Face
+API_KEY = os.environ.get("MY_SECRET_TOKEN", "chave-padrao-muito-segura")
+# O Ollama agora vai rodar internamente na porta 11434
+OLLAMA_URL = "http://127.0.0.1:11434"
+
+@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
+def proxy(path):
+    # Verifica se a chave no cabeçalho é igual à sua chave secreta
+    auth_header = request.headers.get("Authorization")
+    if auth_header != f"Bearer {API_KEY}":
+        return Response("Não autorizado", status=401)
+
+    # Se estiver correto, repassa a mensagem para o Ollama
+    url = f"{OLLAMA_URL}/{path}"
+    resp = requests.request(
+        method=request.method,
+        url=url,
+        headers={k: v for k, v in request.headers if k.lower() != 'host'},
+        data=request.get_data(),
+        cookies=request.cookies,
+        allow_redirects=False,
+        stream=True
+    )
+    return Response(resp.iter_content(chunk_size=1024), resp.status_code, resp.headers.items())
+
+if __name__ == "__main__":
+    # O porteiro escuta na porta 7860 (a que o HF abre para o mundo)
+    app.run(host='0.0.0.0', port=7860)