fix zip archive api response.

main.py

@@ -71,15 +71,15 @@ async def process_video_api(video_file: UploadFile = File(...),
                 finally:
                     srt_file.file.close()
             logging.info("Processing the video...")
-            output_path = process_video(temp_input_path, SRT_PATH, max_words_per_line, fontsize, font, bg_color, text_color)
+            output_path, srt_path = process_video(temp_input_path, SRT_PATH, max_words_per_line, fontsize, font, bg_color, text_color)
             logging.info("Archiving response...")
-            zip_path = zip_response(os.path.join(temp_vid_dir,"archive.zip"), [SRT_PATH, output_path])
-            return FileResponse(zip_path, media_type='application/zip', filename=f'result_{video_file.filename}.zip')
+            zip_path = zip_response(os.path.join(temp_vid_dir,"archive.zip"), [output_path, srt_path])
+            return FileResponse(zip_path, media_type='application/zip', filename=f"result_{video_file.filename.split('.')[0]}.zip")
         logging.info("Processing the video...")
-        output_path = process_video(temp_input_path, None, max_words_per_line, fontsize, font, bg_color, text_color)
+        output_path, srt_path = process_video(temp_input_path, None, max_words_per_line, fontsize, font, bg_color, text_color)
         logging.info("Archiving response...")
-        zip_path = zip_response(os.path.join(temp_vid_dir,"archive.zip"), [SRT_PATH, output_path])
-        return  FileResponse(zip_path, media_type='application/zip', filename=f'result_{video_file.filename}.zip')
+        zip_path = zip_response(os.path.join(temp_vid_dir,"archive.zip"), [output_path, srt_path])
+        return  FileResponse(zip_path, media_type='application/zip', filename=f"result_{video_file.filename.split('.')[0]}.zip")
                 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

utils/process_video.py

@@ -25,12 +25,14 @@ def process_video(invideo_filename:str,
     if srt_path:
         subtitler(invideo_filename, srt_path, OUTVIDEO_PATH, fontsize, font, bg_color, text_color)
         return OUTVIDEO_PATH
+    logging.info("Converting Video to Audio")
     INAUDIO_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.mp3")
-    convert_mp4_to_mp3(invideo_filename, INAUDIO_PATH)
+    if not os.path.exists(INAUDIO_PATH):
+        convert_mp4_to_mp3(invideo_filename, INAUDIO_PATH)
     SRT_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.srt") 
+    logging.info("Transcribing...")
     if not os.path.exists(SRT_PATH):
         transcriber(INAUDIO_PATH, SRT_PATH, max_words_per_line)
-    logging.info("Transcription Complete")
+    logging.info("Subtitling...")
     subtitler(invideo_filename, SRT_PATH, OUTVIDEO_PATH, fontsize, font, bg_color, text_color)
-    logging.info("Subtitling Complete")
-    return OUTVIDEO_PATH
+    return OUTVIDEO_PATH, SRT_PATH

utils/zip_response.py

@@ -6,5 +6,4 @@ def zip_response(archive_name: str, files: list):
     # Add specific files to the zip file
         for file in files:
             zipf.write(file, arcname=os.path.basename(file))
-            zipf.write(file, arcname=os.path.basename(file))
     return archive_name