From d968e06a9d6c3c03cc74e9b4c1518c94ceab3676 Mon Sep 17 00:00:00 2001
From: AjaySi <ajay.calsoft@gmail.com>
Date: Tue, 9 Apr 2024 12:30:04 +0530
Subject: [PATCH] Keyword, Audio to Blog - WIP

---
 README.md                                         |  2 +-
 .../audio_to_text_generation/stt_audio_blog.py    | 15 ++++++++++++---
 .../text_generation/main_text_generation.py       |  2 +-
 requirements.txt                                  |  1 +
 4 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 1314629b..f7fecf1b 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-# AI Blog Creation and Management Toolkit
+# AI Blog Content Generation Toolkit - Alwrity
 ![](https://github.com/AJaySi/AI-Blog-Writer/blob/main/workspace/keyword_blog.gif)
 
 ## Introduction
diff --git a/lib/gpt_providers/audio_to_text_generation/stt_audio_blog.py b/lib/gpt_providers/audio_to_text_generation/stt_audio_blog.py
index 3b403b27..626c6f67 100644
--- a/lib/gpt_providers/audio_to_text_generation/stt_audio_blog.py
+++ b/lib/gpt_providers/audio_to_text_generation/stt_audio_blog.py
@@ -49,7 +49,11 @@ def speech_to_text(video_url, output_path='.'):
             logger.info(f"Downloading audio for: {yt.title}")
             global progress_bar
             progress_bar = tqdm(total=1.0, unit='iB', unit_scale=True, desc=yt.title)
-            audio_file = audio_stream.download(output_path)
+            try:
+                audio_file = audio_stream.download(output_path)
+            except Exception as err:
+                logger.error(f"Failed to download audio file: {audio_file}")
+
             progress_bar.close()
             logger.info(f"Audio downloaded: {yt.title} to {output_path}")
         # Audio filepath from local directory.
@@ -78,7 +82,7 @@ def speech_to_text(video_url, output_path='.'):
                 file=open(audio_file, "rb"),
                 response_format="text"
             )
-            logger.info(f"\nYouTube video transcription:\n\n{transcript}\n")
+            logger.info(f"\nYouTube video transcription:\n{yt.title}\n{transcript}\n")
             return transcript, yt.title
 
         except Exception as e:
@@ -90,9 +94,14 @@ def speech_to_text(video_url, output_path='.'):
         sys.exit("Video processing failure.")
 
     finally:
-        if os.path.exists(audio_file):
+        try:
+            if os.path.exists(audio_file):
             os.remove(audio_file)
             logger.info("Temporary audio file removed.")
+        except PermissionError:
+            logger.error(f"Permission error: Cannot remove '{audio_file}'. Please make sure you have the necessary permissions.")
+        except Exception as e:
+            logger.error(f"An error occurred removing audio file: {e}")
 
 
 def long_video(temp_file_name):
diff --git a/lib/gpt_providers/text_generation/main_text_generation.py b/lib/gpt_providers/text_generation/main_text_generation.py
index 5a2a9508..ee5c3366 100644
--- a/lib/gpt_providers/text_generation/main_text_generation.py
+++ b/lib/gpt_providers/text_generation/main_text_generation.py
@@ -32,7 +32,7 @@ def llm_text_gen(prompt):
         # Check if API key is provided for the given gpt_provider
         get_api_key(gpt_provider)
 
-        logger.info(f"Model: {model}, Temp: {temperature}, MaxTokens: {max_tokens}, TopP: {top_p}, N: {n}, FrequencyPenalty: {fp}")
+        logger.info(f"Temp: {temperature}, MaxTokens: {max_tokens}, TopP: {top_p}, N: {n}, FrequencyPenalty: {fp}")
         # Perform text generation using the specified LLM parameters and prompt
         if 'google' in gpt_provider.lower():
             try:
diff --git a/requirements.txt b/requirements.txt
index 867554cc..9b8687b1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -23,3 +23,4 @@ wordcloud
 nltk
 prompt_toolkit
 ipython
+html2image