Create server.py

server.py

@@ -0,0 +1,97 @@
+import os
+import random
+import logging
+from flask import Flask, request, jsonify
+import google.generativeai as genai
+from flask_cors import CORS
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+app = Flask(__name__)
+CORS(app)
+
+# Cấu hình Logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# --- CẤU HÌNH TỪ BIẾN MÔI TRƯỜNG ---
+# 1. API KEYS
+api_keys_env = os.getenv("GEMINI_API_KEYS", "")
+API_KEYS = [k.strip() for k in api_keys_env.split(",") if k.strip()]
+
+if not API_KEYS:
+    logger.warning("WARN: Chưa tìm thấy GEMINI_API_KEYS! Server sẽ không hoạt động đúng.")
+
+# 2. MODELS
+models_env = os.getenv("GEMINI_MODELS", "gemini-2.0-flash-exp,gemini-1.5-pro,gemini-1.5-flash")
+GEMINI_MODELS = [m.strip() for m in models_env.split(",") if m.strip()]
+
+@app.route('/')
+def home():
+    return jsonify({
+        "status": "online",
+        "server": "HT MATH Central Server",
+        "models_available": GEMINI_MODELS,
+        "keys_loaded": len(API_KEYS)
+    })
+
+@app.route('/api/models', methods=['GET'])
+def get_models():
+    """API trả về danh sách model"""
+    return jsonify({"models": GEMINI_MODELS})
+
+@app.route('/api/generate', methods=['POST'])
+def generate_content():
+    if not API_KEYS:
+        return jsonify({'error': 'Server chưa cấu hình API Key'}), 500
+
+    try:
+        data = request.json
+        if not data:
+            return jsonify({'error': 'No data provided'}), 400
+
+        prompt = data.get('prompt', '')
+        image_data = data.get('image')
+        model_name = data.get('model', GEMINI_MODELS[0])
+
+        if not prompt and not image_data:
+            return jsonify({'error': 'Cần cung cấp nội dung hoặc hình ảnh'}), 400
+
+        # --- LOAD BALANCING (Round Robin/Random) ---
+        current_key = random.choice(API_KEYS)
+        genai.configure(api_key=current_key)
+        
+        masked_key = current_key[:5] + "..." + current_key[-3:]
+        logger.info(f"Request -> Model: {model_name} | Key: {masked_key}")
+
+        model = genai.GenerativeModel(model_name)
+        content_parts = [prompt]
+        
+        if image_data:
+            try:
+                import base64
+                from io import BytesIO
+                from PIL import Image
+                image_bytes = base64.b64decode(image_data)
+                image = Image.open(BytesIO(image_bytes))
+                content_parts.append(image)
+            except Exception as e:
+                logger.error(f"Lỗi ảnh: {str(e)}")
+                return jsonify({'error': f'Lỗi xử lý ảnh: {str(e)}'}), 400
+
+        response = model.generate_content(content_parts)
+        
+        if response.text:
+            return jsonify({'result': response.text})
+        else:
+            return jsonify({'result': 'Không có phản hồi từ Gemini'}), 500
+
+    except Exception as e:
+        logger.error(f"Lỗi Server: {str(e)}")
+        return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    # Chạy trên port 7860 (Port chuẩn của Hugging Face)
+    app.run(host='0.0.0.0', port=7860)