update serve

constants.py

@@ -12,7 +12,7 @@ SERVER_PORT = os.getenv("SERVER_PORT", 7860)
 ROOT_PATH = os.getenv("ROOT_PATH", None)
 ELO_RESULTS_DIR = os.getenv("ELO_RESULTS_DIR", "./arena_elo/results/latest")
 
-LOG_SERVER = os.getenv("LOG_SERVER", "https://tigerai.ca")
+LOG_SERVER = os.getenv("LOG_SERVER", None)
 LOG_SERVER_SUBDOAMIN = os.getenv("LOG_SERVER_SUBDIR", "GenAI-Arena-hf-logs")
 LOG_SERVER_ADDR = os.getenv("LOG_SERVER_ADDR", f"{LOG_SERVER}/{LOG_SERVER_SUBDOAMIN}")
 # LOG SERVER API ENDPOINTS

model/model_config.py

@@ -56,11 +56,23 @@ register_model_config(
     online_model=False
 )
 
-# register_model_config(
-#     model_name="wonder3d",
-#     i2s_model=True,
-#     online_model=False
-# )
+register_model_config(
+    model_name="sjc",
+    i2s_model=False,
+    online_model=False
+)
+
+register_model_config(
+    model_name="syncdreamer",
+    i2s_model=True,
+    online_model=False
+)
+
+register_model_config(
+    model_name="wonder3d",
+    i2s_model=True,
+    online_model=False
+)
 
 register_model_config(
     model_name="lgm",

serve/controller.py

@@ -0,0 +1,17 @@
+import argparse
+import os
+import time
+import requests
+import uvicorn
+import numpy as np
+from fastapi import FastAPI, Request
+
+app = FastAPI()
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--host", type=str, default="localhost")
+    parser.add_argument("--port", type=int, default=21001)
+    args = parser.parse_args()
+
+    uvicorn.run(app, host=args.host, port=args.port, log_level="info")

serve/log_server.py

@@ -0,0 +1,55 @@
+from fastapi import FastAPI, File, UploadFile, Form, APIRouter
+from typing import Optional
+import json
+import os
+import aiofiles
+from log_utils import build_logger
+from constants import LOG_SERVER_SUBDOAMIN, APPEND_JSON, SAVE_IMAGE, SAVE_LOG
+
+logger = build_logger("log_server", "log_server.log", add_remote_handler=False)
+
+app = APIRouter(prefix=f"/{LOG_SERVER_SUBDOAMIN}")
+
+@app.post(f"/{APPEND_JSON}")
+async def append_json(json_str: str = Form(...), file_name: str = Form(...)):
+    """
+    Appends a JSON string to a specified file.
+    """
+    # Convert the string back to a JSON object (dict)
+    data = json.loads(json_str)
+    # Append the data to the specified file
+    os.makedirs(os.path.dirname(file_name), exist_ok=True)
+    async with aiofiles.open(file_name, mode='a') as f:
+        await f.write(json.dumps(data) + "\n")
+    
+    logger.info(f"Appended 1 JSON object to {file_name}")
+    return {"message": "JSON data appended successfully"}
+
+@app.post(f"/{SAVE_IMAGE}")
+async def save_image(image: UploadFile = File(...), image_path: str = Form(...)):
+    """
+    Saves an uploaded image to the specified path.
+    """
+    # Note: 'image_path' should include the file name and extension for the image to be saved.
+    os.makedirs(os.path.dirname(image_path), exist_ok=True)
+    async with aiofiles.open(image_path, mode='wb') as f:
+        content = await image.read()  # Read the content of the uploaded image
+        await f.write(content)  # Write the image content to a file
+    logger.info(f"Image saved successfully at {image_path}")
+    return {"message": f"Image saved successfully at {image_path}"}
+
+@app.post(f"/{SAVE_LOG}")
+async def save_log(message: str = Form(...), log_path: str = Form(...)):
+    """
+    Save a log message to a specified log file on the server.
+    """
+    # Ensure the directory for the log file exists
+    if os.path.dirname(log_path):
+        os.makedirs(os.path.dirname(log_path), exist_ok=True)
+    
+    # Append the log message to the specified log file
+    async with aiofiles.open(log_path, mode='a') as f:
+        await f.write(f"{message}\n")
+    
+    logger.info(f"Romote log message saved to {log_path}")
+    return {"message": f"Log message saved successfully to {log_path}"}