shaders?

App/Shaders/Schemas.py

@@ -0,0 +1,11 @@
+from typing import List, Optional
+from pydantic import EmailStr, BaseModel
+from datetime import date, datetime, time, timedelta
+
+
+class BaseRequest(BaseModel):
+    image: str
+
+
+class BaseResponse(BaseModel):
+    result: str

App/Shaders/ShaderRoutes.py

@@ -0,0 +1,56 @@
+from fastapi import APIRouter, status
+from .Schemas import BaseRequest, BaseResponse
+from fastapi import FastAPI, HTTPException, Request
+import os
+
+# from HuggingChat import getChatBot
+
+
+import aiofiles
+from fastapi.responses import StreamingResponse, FileResponse
+from .utils import make_effect
+
+shader_router = APIRouter(tags=["Shaders"])
+
+
+@shader_router.get("/3d-effect")
+async def shader_3d(image_link: str):
+    return make_effect(image_link=image_link)
+
+
+@shader_router.get("/shaderOuput/{audio_name}")
+async def serve_audio(request: Request, audio_name: str):
+    audio_directory = "/tmp/Video"
+    audio_path = os.path.join(audio_directory, audio_name)
+    if not os.path.isfile(audio_path):
+        raise HTTPException(status_code=404, detail="Audio not found")
+
+    range_header = request.headers.get("Range", None)
+    audio_size = os.path.getsize(audio_path)
+
+    if range_header:
+        start, end = range_header.strip().split("=")[1].split("-")
+        start = int(start)
+        end = audio_size if end == "" else int(end)
+
+        headers = {
+            "Content-Range": f"bytes {start}-{end}/{audio_size}",
+            "Accept-Ranges": "bytes",
+            # Optionally, you might want to force download by uncommenting the next line:
+            # "Content-Disposition": f"attachment; filename={audio_name}",
+        }
+
+        content = read_file_range(audio_path, start, end)
+        return StreamingResponse(content, media_type="video/mp4", headers=headers)
+
+    return FileResponse(audio_path, media_type="video/mp4")
+
+
+async def read_file_range(path, start, end):
+    async with aiofiles.open(path, "rb") as file:
+        await file.seek(start)
+        while True:
+            data = await file.read(1024 * 1024)  # read in chunks of 1MB
+            if not data or await file.tell() > end:
+                break
+            yield data

App/Shaders/utils.py

@@ -0,0 +1,22 @@
+import os
+
+os.environ["WINDOW_BACKEND"] = "headless"  # Use software rendering
+os.environ["SKIP_TORCH"] = "1"
+
+from DepthFlow import DepthFlowScene
+import uuid
+
+
+depthflow = DepthFlowScene()
+
+
+def make_effect(image_link):
+    filename = f"{str(uuid.uuid4())}.mp4"
+    destination = os.path.join("/tmp/Video", filename)
+    depthflow.input(image=image_link)
+    depthflow.main(
+        fps=30,
+        output=destination,
+        quality=1,
+    )
+    return {"file": filename}

Dockerfile

@@ -11,7 +11,17 @@ RUN chmod 755 /srv
 
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    aria2 ffmpeg
+    aria2 ffmpeg     libgl1-mesa-dev \
+    libgles2-mesa-dev \
+    libglu1-mesa-dev \
+    freeglut3-dev \
+    libglfw3-dev \
+    libglew-dev \
+    libsdl2-dev \
+    libjpeg-dev \
+    libpng-dev \
+    libfreetype6-dev \
+    && rm -rf /var/lib/apt/lists/*
 
 #copy requirements    
 COPY requirements.txt .

requirements.txt

@@ -24,7 +24,18 @@ bs4
 strenum 
 pydub
 aiofiles
+PyOpenGL 
+PyOpenGL_accelerate 
+glfw 
+Pillow 
+numpy
+broken-source
 
 
 git+https://github.com/snowby666/poe-api-wrapper.git@2c91207598ad930901cb9fb09734705056b7b6a9
-git+https://github.com/Mbonea-Mjema/ballyregan.git
+git+https://github.com/Mbonea-Mjema/ballyregan.git
+
+--find-links https://download.pytorch.org/whl/cpu
+torch
+torchvision
+torchaudio