move config to variables and secrets

README.md

@@ -4,7 +4,7 @@ emoji: 🐢
 colorFrom: purple
 colorTo: green
 sdk: docker
-pinned: false
+pinned: true
 license: cc-by-nc-4.0
 ---
 

main.py

@@ -12,7 +12,6 @@ from pydantic import BaseModel, field_validator
 
 app = FastAPI()
 security = HTTPBasic()
-api_configs_file = os.path.abspath("api_config.yml")
 
 class MP4Video(BaseModel):
     video_file: UploadFile
@@ -106,7 +105,7 @@ async def process_video_api(video_file: MP4Video = Depends(),
                 return Response(content = zip_file)
             with open(os.path.join(temp_dir, f"{video_file.filename.split('.')[0]}.srt"), 'w+b') as temp_srt_file:
                 logging.info("Processing the video...")
-                output_path, srt_path = process_video(temp_file.name, None, task, max_words_per_line, fontsize, font, bg_color, text_color, caption_mode, api_configs_file)
+                output_path, srt_path = process_video(temp_file.name, None, task, max_words_per_line, fontsize, font, bg_color, text_color, caption_mode)
                 logging.info("Zipping response...")
                 with open(os.path.join(temp_dir, f"{video_file.filename.split('.')[0]}.zip"), 'w+b') as temp_zip_file:
                     zip_file = zip_response(temp_zip_file.name, [output_path, srt_path])

utils/api_configs.py

@@ -1,6 +0,0 @@
-import yaml
-
-def api_configs(config_file):
-    with open(config_file, 'r') as f:
-        db_config = yaml.safe_load(f)
-    return db_config["api_config"]

utils/transcriber.py

@@ -1,11 +1,10 @@
+import os
 from gradio_client import Client, handle_file
-from utils.api_configs import api_configs
 
 def transcriber(invideo_file:str, srt_file:str,
-        max_words_per_line:int, task:str,
-        config_file:str):
-        HF_TOKEN = api_configs(config_file)["secrets"]["hf-token"]
-        HF_SPACE = api_configs(config_file)["secrets"]["hf-space"]
+        max_words_per_line:int, task:str):
+        HF_TOKEN = os.getenv("hf_token")
+        HF_SPACE = os.getenv("hf_space")
         client = Client(HF_SPACE, hf_token=HF_TOKEN)
         result = client.predict(
                 video_input=handle_file(invideo_file),