Update main.py

main.py

@@ -4,7 +4,8 @@ import httpx
 import time
 import base64
 import urllib.parse
-import os # Thêm import os
+import os
+import math # Thêm import math
 from pydantic import BaseModel, Field
 from typing import List, Optional, Union, Literal
 
@@ -18,7 +19,7 @@ class OpenAIImageRequest(BaseModel):
     user: Optional[str] = None
 
     # Pollinations specific parameters
-    seed_value: Optional[int] = Field(default=None, alias="seed", description="Seed for Pollinations. The query param in Pollinations is 'seed'.")
+    seed_value: Optional[int] = Field(default=None, alias="seed", description="Seed provided by the user. Will be transformed before use.")
     model: Optional[str] = Field(default=None, description="Model to use for Pollinations.")
     enhance: Optional[bool] = Field(default=True, description="Enhance parameter for Pollinations.")
     nologo: Optional[bool] = Field(default=True, description="NoLogo parameter for Pollinations.")
@@ -41,8 +42,8 @@ class OpenAIImageResponse(BaseModel):
 # --- FastAPI Application ---
 app = fastapi.FastAPI(
     title="OpenAI-compatible Image Generation API for Pollinations",
-    description="This API wraps the image.pollinations.ai service to provide an OpenAI-like interface.",
-    version="1.0.0"
+    description="This API wraps the image.pollinations.ai service to provide an OpenAI-like interface with custom seed transformation.",
+    version="1.1.0" # Updated version
 )
 
 # --- Helper Functions ---
@@ -77,6 +78,7 @@ async def create_image_generation(request: OpenAIImageRequest):
     """
     Mimics the OpenAI image generation endpoint.
     Receives a prompt and other parameters, then calls the Pollinations API.
+    If a seed is provided by the user, it's transformed before being sent.
     """
     pollinations_base_url = "https://image.pollinations.ai/prompt/"
     results_data: List[Union[ImageURL, ImageB64]] = []
@@ -88,6 +90,24 @@ async def create_image_generation(request: OpenAIImageRequest):
             detail="Invalid 'size' format. Expected 'widthxheight', e.g., '1024x1024'."
         )
 
+    # --- Seed Transformation Logic ---
+    final_seed_to_use = None
+    if request.seed_value is not None:
+        user_provided_seed = float(request.seed_value)
+        current_timestamp_int = int(time.time()) # Thời gian hiện tại (giây)
+        sqrt2_times_pi = math.sqrt(2) * math.pi # √2 × π
+
+        intermediate_sum = user_provided_seed + float(current_timestamp_int) + sqrt2_times_pi
+        final_seed_to_use = math.floor(intermediate_sum) % 10 # Lấy số nguyên hàng đơn vị (0-9)
+
+        # Ghi log để debug (bạn có thể xóa hoặc giữ lại)
+        print(f"User provided seed: {request.seed_value}")
+        print(f"Current timestamp (int): {current_timestamp_int}")
+        print(f"Calculated constant (sqrt(2)*pi): {sqrt2_times_pi}")
+        print(f"Intermediate sum for seed: {intermediate_sum}")
+        print(f"Transformed seed for Pollinations: {final_seed_to_use}")
+    # --- End Seed Transformation Logic ---
+
     async with httpx.AsyncClient() as client:
         for _ in range(request.n):
             encoded_prompt = urllib.parse.quote(request.prompt)
@@ -98,8 +118,12 @@ async def create_image_generation(request: OpenAIImageRequest):
                 query_params["width"] = width
             if height:
                 query_params["height"] = height
-            if request.seed_value is not None:
-                query_params["seed"] = request.seed_value
+            
+            if final_seed_to_use is not None: # Sử dụng seed đã biến đổi nếu có
+                query_params["seed"] = final_seed_to_use
+            # Nếu người dùng không cung cấp seed_value, final_seed_to_use sẽ là None
+            # và không có tham số 'seed' nào được gửi, để Pollinations tự xử lý.
+
             if request.model:
                 query_params["model"] = request.model
             if request.enhance is not None:
@@ -138,11 +162,8 @@ async def create_image_generation(request: OpenAIImageRequest):
 
 # --- Main guard for running with Uvicorn ---
 if __name__ == "__main__":
-    # Khối này dành cho việc thực thi cục bộ (ví dụ: python main.py)
-    # Khi chạy trong Docker trên Hugging Face, CMD trong Dockerfile sẽ được sử dụng.
-    port_to_use = 7860  # Mặc định cho HF Spaces và yêu cầu của bạn
+    port_to_use = 7860
     try:
-        # Hugging Face Spaces có thể đặt biến môi trường PORT
         port_from_env = os.environ.get("PORT")
         if port_from_env:
             port_to_use = int(port_from_env)
@@ -152,5 +173,4 @@ if __name__ == "__main__":
         print(f"Error reading PORT environment variable: {e}. Using default port {port_to_use}.")
         
     print(f"Starting Uvicorn server on host 0.0.0.0, port {port_to_use}")
-    # reload=True hữu ích cho phát triển cục bộ, Docker CMD sẽ không dùng reload.
     uvicorn.run("main:app", host="0.0.0.0", port=port_to_use, reload=True)