Create Dockerfile

Dockerfile

@@ -0,0 +1,79 @@
+FROM alpine:latest
+
+WORKDIR /app
+
+ENV PORT=7860
+ENV MODEL_ID="gemini-1.5-flash-latest"
+ENV GENERATE_CONTENT_API="streamGenerateContent"
+
+RUN apk update && \
+    apk add --no-cache python3 py3-pip curl jq && \
+    rm -rf /var/cache/apk/*
+
+RUN pip3 install --no-cache-dir flask flask-cors
+
+RUN echo '#!/bin/bash \n\
+set -e \n\
+\n\
+SYSTEM_PROMPT="$1" \n\
+USER_PROMPT="$2" \n\
+\n\
+API_KEYS=("apikey1" "apikey2" "apikey3") \n\
+RANDOM_INDEX=$((RANDOM % ${#API_KEYS[@]})) \n\
+GEMINI_API_KEY=${API_KEYS[$RANDOM_INDEX]} \n\
+\n\
+JSON_PAYLOAD=$(jq -n \
+    --arg sp "$SYSTEM_PROMPT" \
+    --arg up "$USER_PROMPT" \
+    \'{ "system_instruction": { "parts": [ { "text": $sp } ] }, "contents": [ { "role": "user", "parts": [ { "text": $up } ] } ] }\') \n\
+\n\
+API_URL="https://generativelanguage.googleapis.com/v1beta/models/${MODEL_ID}:${GENERATE_CONTENT_API}?key=${GEMINI_API_KEY}" \n\
+\n\
+curl -s -X POST -H "Content-Type: application/json" "$API_URL" -d "$JSON_PAYLOAD" | \
+jq -r \'.candidates[0].content.parts[0].text\' | \
+tr -d \'\\n\' \
+' > /usr/local/bin/gemini_handler && chmod +x /usr/local/bin/gemini_handler
+
+RUN echo 'import os \n\
+import subprocess \n\
+from flask import Flask, request, jsonify \n\
+from flask_cors import CORS \n\
+\n\
+app = Flask(__name__) \n\
+CORS(app) \n\
+\n\
+@app.route("/") \n\
+def health_check(): \n\
+    return jsonify({"status": "ok", "message": "API is running"}) \n\
+\n\
+@app.route("/chat", methods=["POST"]) \n\
+def chat_handler(): \n\
+    data = request.get_json() \n\
+    if not data or "prompt" not in data: \n\
+        return jsonify({"error": "Missing \'prompt\' in request body"}), 400 \n\
+\n\
+    system_prompt = data.get("system", "You are a helpful assistant.") \n\
+    user_prompt = data.get("prompt") \n\
+\n\
+    try: \n\
+        result = subprocess.run( \n\
+            ["/usr/local/bin/gemini_handler", system_prompt, user_prompt], \n\
+            capture_output=True, \n\
+            text=True, \n\
+            check=True, \n\
+            timeout=60 \n\
+        ) \n\
+        return jsonify({"response": result.stdout.strip()}) \n\
+    except subprocess.CalledProcessError as e: \n\
+        return jsonify({"error": "Failed to call Gemini API", "details": e.stderr}), 500 \n\
+    except subprocess.TimeoutExpired: \n\
+        return jsonify({"error": "Request to Gemini API timed out"}), 504 \n\
+\n\
+if __name__ == "__main__": \n\
+    port = int(os.environ.get("PORT", 7860)) \n\
+    app.run(host="0.0.0.0", port=port) \n\
+' > app.py
+
+EXPOSE 7860
+
+CMD ["python3", "app.py"]