test

Dockerfile

@@ -2,10 +2,10 @@ FROM python:3.12.2
 
 WORKDIR /code
 
-COPY ./requirements.txt /code/requirements.txt
+COPY ./test_requirements.txt /code/test_requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/test_requirements.txt
 
 COPY . .
 
-CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]
+CMD ["gunicorn","-b", "0.0.0.0:7860", "test:app"]

test.py

@@ -0,0 +1,42 @@
+# ===東吳大學資料系 2025 年 LINEBOT ===
+from google import genai
+from flask import Flask, send_from_directory
+import os
+import logging
+import tempfile
+
+# === 初始化 Google Gemini ===
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+client = genai.Client(api_key=GOOGLE_API_KEY)
+MODEL_ID = "gemini-2.0-flash"
+
+# === 初始設定 ===
+static_tmp_path = tempfile.gettempdir()
+os.makedirs(static_tmp_path, exist_ok=True)
+base_url = os.getenv("SPACE_HOST")  # e.g., "your-space-name.hf.space"
+
+# === Flask 應用初始化 ===
+app = Flask(__name__)
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
+app.logger.setLevel(logging.INFO)
+
+# === AI Query 包裝 ===
+def query(payload):
+    response = client.models.generate_content(
+    model=MODEL_ID,
+    contents=payload
+    )
+    return response.text
+
+# === 靜態圖檔路由 ===
+@app.route("/images/<filename>")
+def serve_image(filename):
+    return send_from_directory(static_tmp_path, filename)
+
+
+# === LINE Webhook 接收端點 ===
+@app.route("/")
+def home():
+    return {"message": query("用繁體中文介紹你自己")}

test_requirements.txt

@@ -0,0 +1,7 @@
+Flask
+gunicorn
+line-bot-sdk
+markdown
+beautifulsoup4
+openai
+google-genai