Rename app/proxy.py to app.py

app.py

@@ -0,0 +1,33 @@
+from flask import Flask, request, jsonify
+import os
+import requests
+
+app = Flask(__name__)
+
+# Load the OpenAI API key from environment variables
+API_KEY = os.getenv("OPENAI_API_KEY")
+BASE_URL = "https://api.typegpt.net/v1"
+
+# A simple proxy to OpenAI's endpoints
+@app.route("/<path:endpoint>", methods=["POST"])
+def proxy(endpoint):
+    # Check for a fake API key from the client
+    client_key = request.headers.get("Authorization")
+    if client_key != "Bearer fake-key":
+        return jsonify({"error": "Unauthorized"}), 401
+
+    # Forward the request to OpenAI with the actual API key
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json"
+    }
+    data = request.get_json()
+
+    # Make the request to OpenAI API
+    response = requests.post(f"{BASE_URL}/{endpoint}", headers=headers, json=data)
+
+    # Forward OpenAI's response back to the client
+    return jsonify(response.json()), response.status_code
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=5000)

app/proxy.py

@@ -1,71 +0,0 @@
-import os
-import requests
-from dotenv import load_dotenv
-from flask import Flask, request, jsonify, Response
-
-# Load environment variables from .env file
-load_dotenv()
-
-app = Flask(__name__)
-
-# OpenAI API base URL
-OPENAI_API_BASE_URL = "https://api.typegpt.net/v1"
-
-# Retrieve API keys from environment variables
-OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
-APP_API_KEY = os.getenv('APP_API_KEY')
-
-if not OPENAI_API_KEY:
-    raise ValueError("OpenAI API key not found. Please set OPENAI_API_KEY in the .env file.")
-
-if not APP_API_KEY:
-    raise ValueError("App API key not found. Please set APP_API_KEY in the .env file.")
-
-@app.route('/v1/chat/completions', methods=['POST'])
-def chat_completions():
-    """
-    Proxy endpoint for OpenAI's /v1/chat/completions.
-    Requires APP_API_KEY for authentication.
-    """
-    # Authenticate the request
-    client_api_key = request.headers.get('Authorization')
-    if not client_api_key:
-        return jsonify({'error': 'Authorization header missing.'}), 401
-
-    if client_api_key != f"Bearer {APP_API_KEY}":
-        return jsonify({'error': 'Invalid API key.'}), 403
-
-    # Forward the request to OpenAI
-    openai_url = f"{OPENAI_API_BASE_URL}/chat/completions"
-    headers = {
-        'Content-Type': 'application/json',
-        'Authorization': f'Bearer {OPENAI_API_KEY}'
-    }
-
-    try:
-        # Forward the JSON payload
-        payload = request.get_json()
-
-        if not payload:
-            return jsonify({'error': 'Invalid JSON payload.'}), 400
-
-        # Make the POST request to OpenAI
-        response = requests.post(openai_url, headers=headers, json=payload)
-
-        # Forward OpenAI's response back to the client
-        return Response(
-            response.content,
-            status=response.status_code,
-            content_type=response.headers.get('Content-Type', 'application/json')
-        )
-
-    except requests.exceptions.RequestException as e:
-        return jsonify({'error': str(e)}), 500
-
-@app.route('/')
-def index():
-    return "Secure OpenAI API Proxy - /v1/chat/completions Only"
-
-if __name__ == '__main__':
-    # It's recommended to use a production-ready server like Gunicorn for deployment
-    app.run(host='0.0.0.0', port=5000)