api.py

import os
from fastapi import FastAPI
from pydantic import BaseModel
import boto3
import botocore
from fastapi.middleware.cors import CORSMiddleware
# from pydantic_settings import BaseSettings

# class Settings(BaseSettings):
#     AWS_ACCESS_KEY_ID: str
#     AWS_SECRET_ACCESS_KEY: str

# settings = Settings()

app = FastAPI()

# Configure CORS settings
origins = [
    "http://localhost:3000",
      "http://localhost:3000/input"  # Add your frontend origin(s) here
    # Add more origins if needed
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],  # You can restrict HTTP methods if needed
    allow_headers=["*"],  # You can restrict headers if needed
)
# AWS_ACCESS_KEY_ID = settings.AWS_ACCESS_KEY_ID
# AWS_SECRET_ACCESS_KEY = settings.AWS_SECRET_ACCESS_KEY
# print(AWS_ACCESS_KEY_ID , AWS_SECRET_ACCESS_KEY)

class InputParameters(BaseModel):
    phase: str
    config_file: str
    config_spec: str
    vid_dir: str
    frame_ext: str
    out_dir: str
    amplification_factor: int
    velocity_mag: bool
    fl: float
    fh: float
    fs: float
    n_filter_tap: int
    filter_type: str
    Temporal: bool

class JsonRequest(BaseModel):
    selectedVideo: str
    inputParameters: InputParameters


def upload_file_to_s3(file_name,bucket,object_name):
    print("Uploading video to S3...")
    s3 = boto3.client('s3',aws_access_key_id='',aws_secret_access_key='')

    if object_name is None:
        object_name = os.path.basename(file_name)
    try:
        s3.upload_file(file_name, bucket, object_name, ExtraArgs={'ContentType': 'video/mp4'})
        print(f"File '{file_name}' uploaded to S3 bucket '{bucket}' as '{object_name}'")
        return True
    except Exception as e:
        print(f"Error uploading file to S3: {str(e)}")
        return False

def download_video_from_s3(bucket_name,key, download_path):
    print("Downloading video from S3...")
    s3 = boto3.resource('s3',aws_access_key_id='',aws_secret_access_key='')
    try:
        s3.Bucket(bucket_name).download_file(key, download_path)
    except botocore.exceptions.ClientError as e:
        if e.response['Error']['Code'] == "404":
            print("The object does not exist.")
        else:
            raise

@app.post("/upload/")
async def get(json: JsonRequest):
    BUCKET_NAME = "skillissuevid"
    print("json:" ,  json)
    obj = json.selectedVideo.split("/")[-1]
    download_video_from_s3(BUCKET_NAME,obj,obj)
    name = obj.split(".")[0]
    if not os.path.exists(f"data/vids/{name}"):
        os.mkdir(f"data/vids/{name}")
    os.system(f"ffmpeg -i {obj} data/vids/{name}"+"/%06d.png")
    if json.inputParameters.Temporal:
        command = (
        f"python main.py --config_file=configs/o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3.conf --phase=run_temporal --vid_dir=data/vids/{name} --out_dir=data/output/{name}_o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3 --amplification_factor={int(json.inputParameters.amplification_factor)} --fl={float(json.inputParameters.fl)} --fh={float(json.inputParameters.fh)} --fs={int(json.inputParameters.fs)} --n_filter_tap={int(json.inputParameters.n_filter_tap)} --filter_type={json.inputParameters.filter_type}")
        folder = f"{name}_o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3_fl{json.inputParameters.fl}_fh{json.inputParameters.fh}_fs{json.inputParameters.fs}_n{json.inputParameters.n_filter_tap}_{json.inputParameters.filter_type}"
    else:
        command = (
        f"python main.py  --config_file=configs/o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3.conf --phase=run --vid_dir=data/vids/{name} --out_dir=data/output/{name}_o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3 --amplification_factor={int(json.inputParameters.amplification_factor)}"
        )
        folder=f"{name}_o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3"
    os.system(command)
    upload_file_to_s3(f"data/output/{folder}/{folder}_259002.mp4",BUCKET_NAME,f"{name}_output.mp4")
    return {"link": f"https://d175wanlbunlv0.cloudfront.net/{name}_output.mp4","inputParameters":json.inputParameters}



if __name__ == '__main__':
    print("Starting server...")
    os.system("uvicorn api:app --host localhost --port 8080")