yt_router (untested)

App/Youtube/Schema.py

@@ -0,0 +1,31 @@
+from pydantic import BaseModel, Field, HttpUrl
+
+
+class YouTubeUploadTask(BaseModel):
+    filename: HttpUrl = Field(..., description="URL of the video file to upload")
+    title: str = Field(
+        ...,
+        min_length=100,
+        max_length=500,
+        description="A good title for the video",
+    )
+    description: str = Field(
+        ...,
+        min_length=100,
+        max_length=500,
+        description="A brief summary of the video's content",
+    )
+    category_id: str = "22"  # Default to a generic category, update as needed
+    privacy: str = "private"
+    tags: str = Field(
+        ...,
+        min_length=100,
+        max_length=500,
+        description="Best seo tags for youtube based on the story",
+    )
+    thumbnail: str = Field(
+        ...,
+        min_length=100,
+        max_length=500,
+        description="""Best image prompt based on the image description: here is an """,
+    )

App/Youtube/youtubeRouter.py

@@ -0,0 +1,110 @@
+from fastapi import APIRouter, HTTPException, BackgroundTasks
+from subprocess import run
+import tempfile
+import os
+import logging
+import aiohttp
+import asyncio
+from .Schema import YouTubeUploadTask
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/youtube", tags=["youtube"])
+
+
+async def download_file(url: str) -> str:
+    """Download file from URL and save to temp directory."""
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as response:
+                if response.status != 200:
+                    raise HTTPException(
+                        status_code=400,
+                        detail=f"Failed to download file: HTTP {response.status}",
+                    )
+
+                # Create temp file with appropriate extension
+                content_type = response.headers.get("content-type", "")
+                ext = ".mp4" if "video" in content_type else ".tmp"
+
+                fd, temp_path = tempfile.mkstemp(suffix=ext)
+
+                # Write content to temp file
+                with os.fdopen(fd, "wb") as tmp:
+                    while chunk := await response.content.read(8192):
+                        tmp.write(chunk)
+
+                return temp_path
+    except Exception as e:
+        logger.error(f"Error downloading file: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"File download failed: {str(e)}")
+
+
+async def upload_video_background(task: YouTubeUploadTask):
+    """Background task to handle video upload."""
+    temp_file = None
+    try:
+        logger.info(f"Starting download for video: {task.filename}")
+        temp_file = await download_file(task.filename)
+        logger.info(f"Download complete. Saved to: {temp_file}")
+
+        # Build the command
+        command = [
+            "/srv/youtube/youtubeuploader",
+            "-filename",
+            temp_file,
+            "-title",
+            task.title,
+            "-description",
+            task.description,
+            "-categoryId",
+            task.category_id,
+            "-privacy",
+            task.privacy,
+            "-tags",
+            task.tags,
+        ]
+
+        if task.thumbnail:
+            command.extend(["-thumbnail", task.thumbnail])
+
+        logger.info("Executing upload command")
+        result = run(command, capture_output=True, text=True)
+
+        if result.returncode != 0:
+            logger.error(f"Upload failed: {result.stderr}")
+            raise Exception(f"Upload failed: {result.stderr}")
+
+        logger.info("Upload completed successfully")
+        logger.debug(f"Upload output: {result.stdout}")
+
+    except Exception as e:
+        logger.error(f"Error in upload process: {str(e)}")
+        raise
+    finally:
+        # Clean up temp file
+        if temp_file and os.path.exists(temp_file):
+            try:
+                os.remove(temp_file)
+                logger.info(f"Cleaned up temporary file: {temp_file}")
+            except Exception as e:
+                logger.error(f"Failed to clean up temp file: {str(e)}")
+
+
+@router.post("/upload")
+async def upload_video_to_youtube(
+    task: YouTubeUploadTask, background_tasks: BackgroundTasks
+):
+    """
+    Endpoint to handle YouTube video upload requests.
+    The actual upload is performed as a background task.
+    """
+    try:
+        background_tasks.add_task(upload_video_background, task)
+        return {"message": "Upload task started", "status": "processing"}
+
+    except Exception as e:
+        logger.error(f"Error initiating upload task: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))

App/app.py

@@ -5,7 +5,7 @@ from fastapi.middleware.gzip import GZipMiddleware
 from .TTS.TTSRoutes import tts_router
 from .Embedding.EmbeddingRoutes import embeddigs_router
 from .OCR.Tesseract import tessaract_ocr_router
-
+from .Youtube.youtubeRouter import router as yt_router
 from fastapi.middleware.cors import CORSMiddleware
 
 from fastapi_cache import FastAPICache
@@ -48,4 +48,5 @@ async def landing_page():
 app.include_router(embeddigs_router)
 app.include_router(tessaract_ocr_router)
 app.include_router(tts_router)
+app.include_router(yt_router)
 # app.include_router(shader_router)

Dockerfile

@@ -40,6 +40,13 @@ USER admin
 
 COPY --chown=admin . /srv
 
+# Install youtubeuploader
+ADD https://github.com/porjo/youtubeuploader/releases/download/23.06/youtubeuploader_23.06_Linux_x86_64.tar.gz youtubeuploader.tar.gz
+RUN mkdir -p /srv/youtube && \
+  tar -zxvf youtubeuploader.tar.gz -C /srv/youtube && \
+  rm youtubeuploader.tar.gz && \
+  chmod +x /srv/youtube/youtubeuploader
+
 # Command to run the application
 CMD uvicorn App.app:app --host 0.0.0.0 --port 7860 --workers 1
 # Expose the server port