Added Audio Download in Transcription

main.py

@@ -125,13 +125,23 @@ async def download_media(user_id: str, media_type: str, media_format: str, media
 @app.get("/get_transcript")
 async def get_transcript(user_id: str, subtitle_format: str = 'srt', word_level: bool = False):
     
-    # Retrieving the media_path from the context for this user's session
-    media_path = users_context[user_id]['media_path']
+    # If Video Already Downloaded
+    if 'media_path' in users_context[user_id].keys():
 
-    # Checking if the media_type is Video, then extract it's audio
-    media_type = users_context[user_id]['media_type']
-    if media_type == 'video':
-        media_path = extract_audio(media_path)
+        # Retrieving the media_path from the context for this user's session
+        media_path = users_context[user_id]['media_path']
+
+        # Checking if the media_type is Video, then extract it's audio
+        media_type = users_context[user_id]['media_type']
+        if media_type == 'video':
+            media_path = extract_audio(media_path)
+    
+    else:
+
+        # Downloading Audio for Transcription
+        media_path = users_context[user_id]['downloader'].download('audio', 'mp3', '128kbps')
+
+        # Retrieving Audio
 
     # # Whisper based transcription
     # stable_whisper_transcript = StableWhisper(media_path, output_folder, subtitle_format=subtitle_format, word_level=word_level)
@@ -195,11 +205,10 @@ async def get_translation(user_id: str, target_language: str = 'en'):
 
 @app.get("/get_summary")
 async def get_summary(user_id: str, Summary_type: str, Summary_strategy: str, Target_Person_type: str,
-                      Response_length:  str, Writing_style: str, text_input: str = None):
+                      Response_length:  str, Writing_style: str):
 
-    # Getting Transcript if not provided
-    if not text_input:
-        text_input = users_context[user_id]['transcript']
+    # Getting Transcript
+    text_input = users_context[user_id]['transcript']
     
     # # Extracting Summary
     # summary_extractor = Extract_Summary(text_input=text_input)
@@ -228,11 +237,10 @@ async def get_summary(user_id: str, Summary_type: str, Summary_strategy: str, Ta
 
 @app.get("/get_key_info")
 async def get_key_info(user_id: str, Summary_type: str, Summary_strategy: str, Target_Person_type: str, 
-                       Response_length: str, Writing_style: str, text_input: str = None):
+                       Response_length: str, Writing_style: str):
 
-    # Getting Transcript if not provided
-    if not text_input:
-        text_input = users_context[user_id]['transcript']
+    # Getting Transcript
+    text_input = users_context[user_id]['transcript']
     
     # # Extracting Summary
     # summary_extractor = Extract_Summary(text_input=text_input)
@@ -260,12 +268,11 @@ async def get_key_info(user_id: str, Summary_type: str, Summary_strategy: str, T
 
 
 @app.get("/get_audiobook")
-async def get_audiobook(user_id: str, narration_style: str, speaker: str = "male", text_input: str = None, 
+async def get_audiobook(user_id: str, narration_style: str, speaker: str = "male", 
                         audio_format: str = "mp3", audio_quality: str = "128kbps"):
 
-    # Getting Transcript if not provided
-    if not text_input:
-        text_input = users_context[user_id]['transcript']
+    # Getting Transcript
+    text_input = users_context[user_id]['transcript']
     
     # # Extracting Narration