Adding files

Dockerfile

@@ -0,0 +1,19 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+# Expose the port the app will run on
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,41 @@
+from fastapi import FastAPI, BackgroundTasks, HTTPException
+from pydantic import BaseModel
+import os
+
+app = FastAPI()
+
+# Define the request body for the training job
+class TrainingJob(BaseModel):
+    code: str
+    token: str
+
+# This is where you'll put your training logic
+def run_training(user_code: str):
+    """
+    This function runs the user's code.
+     In a real application, you would add more sophisticated logic 
+    here,
+    such as saving the results to a file or pushing them to a Hub 
+    model repo.
+    """
+    print("--- Starting Training ---")
+    try:
+        # WARNING: Running exec() on user-provided code is a security risk.
+        # Make sure you understand the implications.
+        exec(user_code)        
+        print("--- Training Finished ---")
+    except Exception as e:
+       print(f"--- Training Failed: {e} ---")
+
+@app.post("/train")
+def start_training(job: TrainingJob, background_tasks: BackgroundTasks):
+    
+    # Authenticate the request
+    if job.token != os.environ.get("AUTH_TOKEN"):
+        raise HTTPException(status_code=401, detail="Unauthorized"
+    )
+
+    # Start the training in the background
+    background_tasks.add_task(run_training, job.code)
+
+    return {"message": "Training job started successfully!"}

requirements.txt

@@ -0,0 +1,9 @@
+fastapi
+uvicorn[standard]
+stable-baselines3[extra]
+gymnasium[all]
+pettingzoo[all]
+torch
+huggingface_hub
+pydantic==2.12.4
+pydantic_core==2.41.5