cicd for hf

.github/workflows/push_to_hf.yaml

@@ -0,0 +1,37 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [main]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+
+      - name: Clean cache files
+        run: |
+          find . -type d -name "__pycache__" -exec rm -rf {} +
+          find . -type f -name "*.pyc" -delete
+          
+      - name: Install Git LFS
+        run: |
+          git lfs install
+          git lfs track "*.png"
+          git lfs track "*.jpg"
+          git lfs track "*.mp4"
+          git lfs track "*.pt"
+          git add .gitattributes
+          git commit -m "Track binaries with LFS" || echo "No changes"
+
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push --force https://DvorakInnovationAI:$HF_TOKEN@huggingface.co/spaces/DvorakInnovationAI/REVEL-RIZZ-API main
+

.gitignore

@@ -1,4 +1,5 @@
 .env
 myenv
 *.pyc
-__pycache__/
+__pycache__/
+delete_pycache.py

Dockerfile

@@ -0,0 +1,39 @@
+# Use an official Python image
+FROM python:3.12-slim
+
+# Create a non-root user with UID 1000
+RUN useradd -m -u 1000 user
+
+# Create a data/cache directory with open permissions
+RUN mkdir -p /data/.cache/huggingface && \
+    chown -R user:user /data && \
+    chmod -R 777 /data
+
+# Set environment variables for Hugging Face cache
+ENV HF_HOME=/data/.cache/huggingface \
+    TRANSFORMERS_CACHE=/data/.cache/huggingface/transformers \
+    HF_DATASETS_CACHE=/data/.cache/huggingface/datasets
+
+# Switch to the non-root user
+USER user
+
+# Set working directory for the user
+WORKDIR /home/user/app
+
+# Ensure pip packages go to user space
+ENV PATH=/home/user/.local/bin:$PATH
+
+# Upgrade pip (runs as user)
+RUN pip install --no-cache-dir --upgrade pip
+
+# Copy code into the container, set ownership
+COPY --chown=user . .
+
+# Install dependencies (will go to ~/.local)
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Expose port
+EXPOSE 7860
+
+# Run the FastAPI app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -0,0 +1,8 @@
+---
+title: revel_rizz
+sdk: docker
+emoji: 🚀
+colorFrom: blue
+app_file: app.py
+pinned: false
+---

api/routers/looks_analyzer.py

@@ -13,7 +13,7 @@ class UserRequest(BaseModel):
 
 
 @router.post("/looks-analyzer")
-async def generate_reply(request: UserRequest):
+async def analyze_looks(request: UserRequest):
     """
     Endpoint to analyze looks
     """

api/routers/pickup_line_generator.py

@@ -16,7 +16,7 @@ class UserRequest(BaseModel):
 
 
 @router.post("/pickup-line-generator")
-async def generate_reply(request: UserRequest):
+async def generate_pickup_line(request: UserRequest):
     """
     Endpoint to extract conversation from a base64-encoded image
     """

app.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI, Request
+from api.routers import reply_generator
+from api.routers import pickup_line_generator
+from api.routers import looks_analyzer
+
+app = FastAPI()
+
+@app.middleware("http")
+async def log_requests(request: Request, call_next):
+    response = await call_next(request)
+    return response
+
+@app.get("/")
+async def root():
+    return {'response':'Hello'}
+
+app.include_router(reply_generator.router, prefix="/api")
+app.include_router(pickup_line_generator.router, prefix="/api")
+app.include_router(looks_analyzer.router, prefix="/api")