Update Dockerfile

Dockerfile

@@ -1,81 +1,95 @@
-FROM python:3.11-alpine
+FROM node:20-slim
 
 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 curl jq bash && \
-    pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir flask flask-cors gunicorn && \
-    rm -rf /var/cache/apk/*
-
-# Metode penulisan file paling robust: echo dengan string kutip tunggal.
-# -e memungkinkan interpretasi \n. Kutip tunggal mencegah SEMUA ekspansi shell.
-RUN echo -e '#!/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 \\\n\
-    --arg sp "$SYSTEM_PROMPT" \\\n\
-    --arg up "$USER_PROMPT" \\\n\
-    '\''{ "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" | \\\n\
-jq -r ".candidates[0].content.parts[0].text" | \\\n\
-tr -d "\\n"\
-' > /usr/local/bin/gemini_handler
-
-RUN chmod +x /usr/local/bin/gemini_handler
-
-# Menulis app.py dengan metode yang sama. Perhatikan semua string di Python menggunakan kutip ganda
-# untuk menghindari konflik dengan kutip tunggal di echo.
-RUN echo -e '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\
-' > app.py
+RUN cat << EOF > package.json
+{
+  "name": "gemini-proxy-server",
+  "version": "1.0.0",
+  "main": "index.js",
+  "scripts": {
+    "start": "node index.js"
+  },
+  "dependencies": {
+    "axios": "^1.7.2",
+    "cors": "^2.8.5",
+    "express": "^4.19.2"
+  }
+}
+EOF
+
+RUN npm install --only=production
+
+RUN cat << EOF > index.js
+const express = require('express');
+const cors = require('cors');
+const axios = require('axios');
+
+const app = express();
+const PORT = 7860;
+const MODEL_ID = "gemini-flash-lite-latest";
+const API_URL_BASE = "https://generativelanguage.googleapis.com/v1beta/models";
+
+const apiKeys = [
+    "apikey1",
+    "apikey2",
+    "apikey3",
+    "apikey4",
+    "apikey5"
+];
+
+const getRandomApiKey = () => {
+    const randomIndex = Math.floor(Math.random() * apiKeys.length);
+    return apiKeys[randomIndex];
+};
+
+app.use(cors());
+app.use(express.json());
+
+app.post('/api/generate', async (req, res) => {
+    const { prompt } = req.body;
+
+    if (!prompt) {
+        return res.status(400).json({ error: 'Request body harus menyertakan "prompt"' });
+    }
+
+    try {
+        const apiKey = getRandomApiKey();
+        const apiUrl = `\${API_URL_BASE}/\${MODEL_ID}:streamGenerateContent?key=\${apiKey}`;
+
+        const requestBody = {
+            contents: [{
+                role: "user",
+                parts: [{ "text": prompt }],
+            }],
+            generationConfig: {
+                thinkingConfig: {
+                    thinkingBudget: 0,
+                },
+            },
+        };
+
+        const geminiResponse = await axios.post(apiUrl, requestBody, {
+            headers: { 'Content-Type': 'application/json' },
+            responseType: 'stream'
+        });
+
+        res.setHeader('Content-Type', geminiResponse.headers['content-type']);
+        geminiResponse.data.pipe(res);
+
+    } catch (error) {
+        console.error("Error saat menghubungi Gemini API:", error.message);
+        const status = error.response ? error.response.status : 500;
+        const data = error.response ? error.response.data : { message: "Internal Server Error" };
+        res.status(status).json(data);
+    }
+});
+
+app.listen(PORT, () => {
+    console.log(`Server berjalan di http://localhost:\${PORT}`);
+});
+EOF
 
 EXPOSE 7860
 
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "app:app"]
+CMD ["npm", "start"]