Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import io
 import json
-from fastapi import FastAPI, File, UploadFile, Form
+from fastapi import FastAPI, File, UploadFile, Form, Request
 from fastapi.responses import HTMLResponse
 from PIL import Image
 import imagehash
@@ -43,6 +43,13 @@ def get_database_hashes():
                 print(f"Error loading {filename}: {e}")
     return hashes
 
+# ฟังก์ชันดึง IP ของผู้ใช้
+def get_client_ip(request: Request):
+    forwarded_for = request.headers.get("x-forwarded-for")
+    if forwarded_for:
+        return forwarded_for.split(",")[0].strip()
+    return request.client.host if request.client else "127.0.0.1"
+
 @app.get("/", response_class=HTMLResponse)
 def serve_index():
     if os.path.exists("index.html"):
@@ -79,7 +86,7 @@ async def check_image(file: UploadFile = File(...)):
             matched_hash_key = uploaded_hash_str
 
         # ดึงคะแนนโหวต
-        image_vote = votes_data.get(matched_hash_key, {"likes": 0, "dislikes": 0})
+        image_vote = votes_data.get(matched_hash_key, {"likes": 0, "dislikes": 0, "users": {}})
         likes = image_vote.get("likes", 0)
         dislikes = image_vote.get("dislikes", 0)
 
@@ -109,11 +116,32 @@ async def check_image(file: UploadFile = File(...)):
         return {"error": f"เกิดข้อผิดพลาดในการประมวลผล: {str(e)}"}
 
 @app.post("/vote")
-async def vote_image(image_hash: str = Form(...), vote_type: str = Form(...)):
+async def vote_image(
+    request: Request,
+    image_hash: str = Form(...),
+    vote_type: str = Form(...),
+    user_id: str = Form(None)
+):
     try:
         votes_data = load_votes()
         if image_hash not in votes_data:
-            votes_data[image_hash] = {"likes": 0, "dislikes": 0}
+            votes_data[image_hash] = {"likes": 0, "dislikes": 0, "users": {}}
+        elif "users" not in votes_data[image_hash]:
+            votes_data[image_hash]["users"] = {}
+
+        # กำหนด Key ระบุตัวตนผู้ใช้ (รวม IP และ User ID)
+        client_ip = get_client_ip(request)
+        user_key = f"{client_ip}_{user_id}" if user_id else client_ip
+
+        # ตรวจสอบว่าผู้ใช้งานคนนี้เคยโหวตรูปนี้ไปแล้วหรือไม่
+        if user_key in votes_data[image_hash]["users"]:
+            return {
+                "success": False,
+                "error": "คุณเคยโหวตรูปภาพนี้ไปแล้ว สามารถโหวตได้เพียง 1 ครั้งเท่านั้น"
+            }
+
+        # บันทึกการโหวตของผู้ใช้งานคนนี้
+        votes_data[image_hash]["users"][user_key] = vote_type
 
         if vote_type == "like":
             votes_data[image_hash]["likes"] += 1