changes

mitul1719 · mitul1719 · commit bbb63b4bc8df · 2025-04-21T16:57:53.000+05:30
diff --git a/app/engine.py b/app/engine.py
@@ -0,0 +1,56 @@
+from fastapi import HTTPException
+from app.utils.ffmpeg_client import run_ffmpeg_command
+from app.config import settings
+
+
+def transcode(input_path, output_path, codec, crf, preset):
+    cmd = ['-y', '-i', input_path, '-c:v', codec, '-preset', preset, '-crf', str(crf), output_path]
+    res = run_ffmpeg_command(cmd)
+    if res.returncode != 0:
+        raise HTTPException(status_code=500, detail=res.stderr.decode())
+    return output_path
+
+
+def measure_quality(reference_path, distorted_path, metrics: list):
+    result = {}
+    if 'vmaf' in metrics:
+        cmd = ['-i', distorted_path, '-i', reference_path, '-lavfi', f"libvmaf=model_path={settings.VMAF_PATH}", '-f', 'null', '-']
+        r = run_ffmpeg_command(cmd)
+        if r.returncode != 0:
+            raise HTTPException(status_code=500, detail=r.stderr.decode())
+        result['vmaf'] = r.stderr.decode()
+    if 'psnr' in metrics:
+        cmd = ['-i', distorted_path, '-i', reference_path, '-lavfi', 'psnr', '-f', 'null', '-']
+        r = run_ffmpeg_command(cmd)
+        result['psnr'] = r.stderr.decode()
+    if 'ssim' in metrics:
+        cmd = ['-i', distorted_path, '-i', reference_path, '-lavfi', 'ssim', '-f', 'null', '-']
+        r = run_ffmpeg_command(cmd)
+        result['ssim'] = r.stderr.decode()
+    return result
+
+
+def process_image(req, input_path, output_path):
+    filters = []
+    for op in req.operations:
+        if op.type == 'resize':
+            filters.append(f"scale={op.params['width']}:{op.params['height']}")
+        if op.type == 'crop':
+            filters.append(f"crop={op.params['width']}:{op.params['height']}:{op.params.get('x',0)}:{op.params.get('y',0)}")
+        if op.type == 'filter':
+            filters.append(f"{op.params['name']}={op.params.get('args','')}")
+        if op.type == 'watermark':
+            filters.append(f"movie={op.params['file']}[wm];[in][wm]overlay={op.params.get('x',10)}:{op.params.get('y',10)}")
+    cmd = ['-y', '-i', input_path, '-vf', ','.join(filters), output_path]
+    res = run_ffmpeg_command(cmd)
+    if res.returncode != 0:
+        raise HTTPException(status_code=500, detail=res.stderr.decode())
+    return output_path
+
+
+def convert_audio(req, input_path, output_path):
+    cmd = ['-y', '-i', input_path, '-c:a', req.target_codec, '-b:a', req.bitrate, '-ar', str(req.sample_rate), '-ac', str(req.channels), output_path]
+    res = run_ffmpeg_command(cmd)
+    if res.returncode != 0:
+        raise HTTPException(status_code=500, detail=res.stderr.decode())
+    return output_path
diff --git a/app/routers/audio.py b/app/routers/audio.py
@@ -0,0 +1,36 @@
+from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
+from app.schemas import ImageReq, JobResp, JobStatusResp
+from app.engine import process_image
+from app.jobs import jobs, Job, JobStatus
+from app.auth import get_current_user
+from app.utils.s3_utils import fetch, upload
+import uuid, time
+
+router = APIRouter(prefix="/api/v1/image", dependencies=[Depends(get_current_user)])
+
+@router.get("/health")
+def health(): return {"status":"ok"}
+
+@router.post("/process", response_model=JobResp)
+def api_process_image(req: ImageReq, background_tasks: BackgroundTasks):
+    job_id=uuid.uuid4().hex; job=Job(job_id); jobs[job_id]=job
+    def run_job():
+        job.status=JobStatus.RUNNING; job.start_time=time.time()
+        try:
+            in_path=fetch(req.input)
+            out_path=req.output.local_path or f"/tmp/{job_id}_out.png"
+            process_image(req, in_path, out_path)
+            upload(out_path, req.output)
+            job.log=f"Image processed to {out_path}"; job.status=JobStatus.SUCCESS
+        except Exception as e:
+            job.error=str(e); job.status=JobStatus.FAILED
+        finally:
+            job.end_time=time.time()
+    background_tasks.add_task(run_job)
+    return {"job_id":job_id}
+
+@router.get("/jobs/{job_id}", response_model=JobStatusResp)
+def get_image_job(job_id):
+    job=jobs.get(job_id)
+    if not job: raise HTTPException(404)
+    return job.to_dict()
diff --git a/app/routers/auth.py b/app/routers/auth.py
@@ -0,0 +1,20 @@
+from fastapi import APIRouter, Depends, HTTPException
+from fastapi.security import OAuth2PasswordRequestForm
+from app.auth import authenticate_user, create_access_token
+from app.schemas import Token
+
+router = APIRouter(prefix="/api/v1/auth")
+
+@router.post("/token", response_model=Token)
+def login(form_data: OAuth2PasswordRequestForm = Depends()):
+    user = authenticate_user(form_data.username, form_data.password)
+    if not user:
+        raise HTTPException(status_code=400, detail="Incorrect username or password")
+    access_token = create_access_token(data={"sub": user["username"]})
+    return {"access_token": access_token, "token_type": "bearer"}
+
+
+
+
+NEW
+
diff --git a/app/routers/image.py b/app/routers/image.py
@@ -0,0 +1,36 @@
+from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
+from app.schemas import ImageReq, JobResp, JobStatusResp
+from app.engine import process_image
+from app.jobs import jobs, Job, JobStatus
+from app.auth import get_current_user
+from app.utils.s3_utils import fetch, upload
+import uuid, time
+
+router = APIRouter(prefix="/api/v1/image", dependencies=[Depends(get_current_user)])
+
+@router.get("/health")
+def health(): return {"status":"ok"}
+
+@router.post("/process", response_model=JobResp)
+def api_process_image(req: ImageReq, background_tasks: BackgroundTasks):
+    job_id=uuid.uuid4().hex; job=Job(job_id); jobs[job_id]=job
+    def run_job():
+        job.status=JobStatus.RUNNING; job.start_time=time.time()
+        try:
+            in_path=fetch(req.input)
+            out_path=req.output.local_path or f"/tmp/{job_id}_out.png"
+            process_image(req, in_path, out_path)
+            upload(out_path, req.output)
+            job.log=f"Image processed to {out_path}"; job.status=JobStatus.SUCCESS
+        except Exception as e:
+            job.error=str(e); job.status=JobStatus.FAILED
+        finally:
+            job.end_time=time.time()
+    background_tasks.add_task(run_job)
+    return {"job_id":job_id}
+
+@router.get("/jobs/{job_id}", response_model=JobStatusResp)
+def get_image_job(job_id):
+    job=jobs.get(job_id)
+    if not job: raise HTTPException(404)
+    return job.to_dict()
diff --git a/app/routers/video.py b/app/routers/video.py
@@ -0,0 +1,43 @@
+from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
+from app.schemas import TranscodeReq, QualityReq, JobResp, JobStatusResp
+from app.engine import transcode, measure_quality
+from app.jobs import jobs, Job, JobStatus
+from app.auth import get_current_user
+from app.utils.s3_utils import fetch, upload
+import uuid, time
+
+router = APIRouter(prefix="/api/v1/video", dependencies=[Depends(get_current_user)])
+
+@router.get("/health")
+def health():
+    return {"status": "ok"}
+
+@router.post("/transcode", response_model=JobResp)
+def api_transcode(req: TranscodeReq, background_tasks: BackgroundTasks):
+    job_id = uuid.uuid4().hex
+    job = Job(job_id); jobs[job_id]=job
+    def run_job():
+        job.status = JobStatus.RUNNING; job.start_time = time.time()
+        try:
+            in_path = fetch(req.input)
+            out_path = req.output.local_path or f"/tmp/{job_id}_out.{req.output.s3_path.split('.')[-1]}"
+            transcode(in_path, out_path, req.codec, req.crf, req.preset)
+            upload(out_path, req.output)
+            job.log = f"Encoded to {out_path}"; job.status = JobStatus.SUCCESS
+        except Exception as e:
+            job.error = str(e); job.status = JobStatus.FAILED
+        finally:
+            job.end_time = time.time()
+    background_tasks.add_task(run_job)
+    return {"job_id": job_id}
+
+@router.get("/jobs/{job_id}", response_model=JobStatusResp)
+def get_job(job_id: str):
+    job = jobs.get(job_id)
+    if not job: raise HTTPException(404, "Job not found")
+    return job.to_dict()
+
+@router.post("/quality")
+def api_quality(req: QualityReq):
+    ref = fetch(req.reference); dist = fetch(req.distorted)
+    return measure_quality(ref, dist, req.metrics)
diff --git a/app/utils/s3_utils.py b/app/utils/s3_utils.py
@@ -0,0 +1,25 @@
+import boto3
+import os
+from urllib.parse import urlparse
+from app.config import settings
+
+s3 = boto3.client('s3',
+    aws_access_key_id=settings.AWS_ACCESS_KEY_ID,
+    aws_secret_access_key=settings.AWS_SECRET_ACCESS_KEY,
+    region_name=settings.AWS_REGION
+)
+
+def fetch(loc):
+    if loc.local_path:
+        return loc.local_path
+    if loc.s3_path:
+        p = urlparse(loc.s3_path)
+        out = f"/tmp/{os.path.basename(p.path)}"
+        s3.download_file(p.netloc, p.path.lstrip('/'), out)
+        return out
+    raise ValueError("Invalid location")
+
+def upload(local, loc):
+    if loc.s3_path:
+        p = urlparse(loc.s3_path)
+        s3.upload_file(local, p.netloc, p.path.lstrip('/'))
