main_config changes - WIP

lib/speech_to_blog/write_blogs_from_youtube_videos.py

@@ -0,0 +1,97 @@
+import os
+import time
+import sys
+
+from pytube import YouTube
+import tempfile
+import openai
+from html2image import Html2Image
+from tqdm import tqdm, trange
+import google.generativeai as genai
+
+from loguru import logger
+logger.remove()
+logger.add(sys.stdout,
+        colorize=True,
+        format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
+    )
+
+from .gpt_providers.stt_audio_blog import speech_to_text
+from .gpt_providers.openai_chat_completion import openai_chatgpt
+
+
+def youtube_to_blog(video_url):
+    """Function to transcribe a given youtube url """
+    # fixme: Doesnt work all types of yt urls.
+    vid_id = video_url.split("=")[1]
+    #hti = Html2Image(output_path="../blog_images")
+    #hti.screenshot(url=video_url, save_as=f"yt-img-{vid_id}.png")
+    #yt_img_path = os.path.join("../blog_images", f"yt-img-{vid_id}.png")
+
+    try:
+        # Starting the speech-to-text process
+        logger.info("Starting with Speech to Text.")
+        audio_text, audio_title = speech_to_text(video_url)
+    except Exception as e:
+        logger.error(f"Error in speech_to_text: {e}")
+        sys.exit(1)  # Exit the program due to error in speech_to_text
+
+    try:
+        # Summarizing the content of the YouTube video
+        audio_blog_content = summarize_youtube_video(audio_text, "gemini")
+        logger.info("Successfully converted given URL to blog article.")
+        return audio_blog_content, audio_title
+    except Exception as e:
+        logger.error(f"Error in summarize_youtube_video: {e}")
+        sys.exit(1)  # Exit the program due to error in summarize_youtube_video
+    return audio_blog_content
+
+
+def summarize_youtube_video(user_content, gpt_providers):
+    """Generates a summary of a YouTube video using OpenAI GPT-3 and displays a progress bar. 
+    Args:
+      video_link: The URL of the YouTube video to summarize.
+    Returns:
+      A string containing the summary of the video.
+    """
+
+    logger.info("Start summarize_youtube_video..")
+    prompt = f"""
+        You are an expert copywriter specializing in digital content writing. I will provide you with a transcript. 
+        Your task is to transform a given transcript into a well-structured and informative blog article. 
+        Please follow the below objectives:
+
+        1. Master the Transcript: Understand main ideas, key points, and the core message.
+        2. Sentence Structure: Rephrase while preserving logical flow and coherence. Dont quote anyone from video.
+        3. Note: Check if the transcript is about programming, then include code examples and snippets in your article.
+        4. Write Unique Content: Avoid direct copying; rewrite in your own words. 
+        5. REMEMBER to avoid direct quoting and maintain uniqueness.
+        6. Proofread: Check for grammar, spelling, and punctuation errors.
+        7. Use Creative and Human-like Style: Incorporate contractions, idioms, transitional phrases, interjections, and colloquialisms.        8. Avoid repetitive phrases and unnatural sentence structures.
+        9. Ensure Uniqueness: Guarantee the article is plagiarism-free.
+        10. Punctuation: Use appropriate question marks at the end of questions.
+        11. Pass AI Detection Tools: Create content that easily passes AI plagiarism detection tools.
+        12. Rephrase words like 'video, youtube, channel' with 'article, blog' and such suitable words.
+
+        Follow the above guidelines to create a well-optimized, unique, and informative article,
+        that will rank well in search engine results and engage readers effectively.
+        Follow above guidelines to craft a blog content from the following transcript:\n{user_content}
+        """
+    if 'gemini' in gpt_providers:
+        try:
+            genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
+        except Exception as err:
+            logger.error("Failed in getting GEMINI_API_KEY")
+        # Use gemini-pro model for text and image.
+        model = genai.GenerativeModel('gemini-pro')
+        try:
+            response = model.generate_content(prompt)
+            return response.text
+        except Exception as err:
+            logger.error("Failed to get response from gemini.")
+    elif 'openai' in gpt_providers:
+        try:
+            response = openai_chatgpt(prompt)
+            return response
+        except Exception as err:
+            SystemError(f"Error in generating blog summary: {err}")