Create app.py

app.py

@@ -0,0 +1,532 @@
+
+
+
+
+
+
+import streamlit as st
+from pydub import AudioSegment
+import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
+import json
+import uuid
+import io
+import edge_tts
+import asyncio
+import aiofiles
+import pypdf
+import os
+import time
+from typing import List, Dict
+
+class PodcastGenerator:
+    def __init__(self):
+        print("PodcastGenerator initialized.")
+
+    async def generate_script(self, prompt: str, language: str, api_key: str) -> Dict:
+        print(f"Generating podcast script with prompt: {prompt[:50]}... and language: {language}")
+
+         
+        # Example structure for the podcast script
+        example = """
+            {
+    "topic": "Zoho Schools",
+    "podcast": [
+        {
+            "speaker": 1,
+            "line": "Okay, so like imagine this. What if there was a place where you didn't have to pay for college?"
+        },
+        {
+            "speaker": 2,
+            "line": "Hmm. Intriguing."
+        },
+        {
+            "speaker": 1,
+            "line": "And on top of that, they, like, actually paid you to go there. Paid you to go to college."
+        },
+        {
+            "speaker": 2,
+            "line": "Okay, now you've got my attention, right?"
+        },
+        {
+            "speaker": 1,
+            "line": "It sounds kind of crazy, but it gets even better. They guarantee you a job at a global tech company after you graduate."
+        },
+        {
+            "speaker": 2,
+            "line": "Wow. Hold on. Tuition free, paid stipend, guaranteed job. There's gotta be a catch."
+        },
+        {
+            "speaker": 1,
+            "line": "I know, that's exactly what we were thinking. It's like someone took every complaint about student debt and useless college courses and just, like, flipped the script."
+        },
+        {
+            "speaker": 2,
+            "line": "So where are we diving in today?"
+        },
+        {
+            "speaker": 1,
+            "line": "We're doing a deep dive into Zoho Schools."
+        },
+        {
+            "speaker": 2,
+            "line": "Zoho Schools?"
+        },
+        {
+            "speaker": 1,
+            "line": "Yeah, these schools created by Zoho, you know, the tech company. And they’re doing things completely differently."
+        },
+        {
+            "speaker": 2,
+            "line": "I'm familiar with Zoho, but this is the first I'm hearing about their schools."
+        },
+        {
+            "speaker": 1,
+            "line": "It's certainly a fascinating case study, right? It really challenges assumptions about the traditional education system, especially for, well, anyone really, but particularly for people looking to break into the tech world."
+        },
+        {
+            "speaker": 2,
+            "line": "So they've got their whole website, right. Like, it's wild. You go on there, and it's like, boom, alternate education universe."
+        },
+        {
+            "speaker": 1,
+            "line": "Okay, I'm pulling it up now. What am I looking for first thing?"
+        },
+        {
+            "speaker": 2,
+            "line": "Right. They don’t just have one Zoho School. They’ve got five different things within Zoho Schools."
+        },
+        {
+            "speaker": 1,
+            "line": "Five different schools?"
+        },
+        {
+            "speaker": 2,
+            "line": "Yeah. Technology, design, business, advanced studies, even graduate studies."
+        },
+        {
+            "speaker": 1,
+            "line": "Wow. So they’ve really got the full spectrum of education covered then."
+        },
+        {
+            "speaker": 2,
+            "line": "Yeah, that’s really interesting. It seems much more thought out than just a quick coding bootcamp or something."
+        },
+        {
+            "speaker": 1,
+            "line": "Right. And get this, it’s in India, too. But they take applications from anywhere in India. So they’re really looking to cultivate talent nationally."
+        },
+        {
+            "speaker": 2,
+            "line": "Yeah, like their own little tech talent incubator."
+        },
+        {
+            "speaker": 1,
+            "line": "That's a great way to put it. So they’re not just looking locally for people who already have some skills. They’re trying to find people with potential all over."
+        },
+        {
+            "speaker": 2,
+            "line": "Exactly. And the thing is, they’ve been doing this for a while."
+        },
+        {
+            "speaker": 1,
+            "line": "Oh yeah, 19 years, over 1600 graduates."
+        },
+        {
+            "speaker": 2,
+            "line": "That’s not like a new experiment."
+        },
+        {
+            "speaker": 1,
+            "line": "Yeah, that’s a pretty solid track record, right? Especially if you consider they make up 15% of Zoho’s workforce."
+        },
+        {
+            "speaker": 2,
+            "line": "Now that’s a serious commitment. It’s not just like, “Oh, look at us. We’re being charitable.”"
+        },
+        {
+            "speaker": 1,
+            "line": "Right. Exactly. They’re putting their money where their mouth is."
+        },
+        {
+            "speaker": 2,
+            "line": "It makes you wonder though, like, how do they actually measure if this is working long-term? You know, do these graduates stay with the company for years and years? Or if they leave, are they competitive in the job market?"
+        },
+        {
+            "speaker": 1,
+            "line": "Those are good questions. That would tell us a lot more, I think, than just the number of graduates."
+        },
+        {
+            "speaker": 2,
+            "line": "Totally. Because they’re so upfront about it. Even on their website, they have this whole thing with Zoho Schools. And they start off with, “How useful was college?”"
+        },
+        {
+            "speaker": 1,
+            "line": "Ooh, that’s a loaded question, isn’t it?"
+        },
+        {
+            "speaker": 2,
+            "line": "I know, right? It’s like they’re already challenging the whole system."
+        },
+        {
+            "speaker": 1,
+            "line": "Yeah, because who hasn’t asked themselves that at some point, right?"
+        },
+        {
+            "speaker": 2,
+            "line": "No, seriously. Especially when you’re, you know, maybe struggling to pay off loans or whatever."
+        },
+        {
+            "speaker": 1,
+            "line": "So they actually asked their employees, “How useful was college?”"
+        },
+        {
+            "speaker": 2,
+            "line": "Oh, really?"
+        },
+        {
+            "speaker": 1,
+            "line": "And most of them were like, “Nah, I learned what I actually use on the job.”"
+        },
+        {
+            "speaker": 2,
+            "line": "Interesting. See that? To me, that really highlights the disconnect between what traditional education is giving us and what employers actually need."
+        },
+        {
+            "speaker": 1,
+            "line": "Yeah, and it’s not just in tech. I mean, we see this everywhere. The skills gap is real."
+        },
+        {
+            "speaker": 2,
+            "line": "It’s like instead of all that theory, they’re like, “Nope, hands-on, let’s go.”"
+        },
+        {
+            "speaker": 1,
+            "line": "Practical experience from day one. Makes sense."
+        },
+        {
+            "speaker": 2,
+            "line": "And the curriculum is always changing, keeping up with all the latest stuff."
+        },
+        {
+            "speaker": 1,
+            "line": "And the teachers aren’t like professors who’ve never left a classroom. They’re pulling in actual professionals."
+        },
+        {
+            "speaker": 2,
+            "line": "Yeah, people from the industry. So they know what actually works, what you actually need."
+        },
+        {
+            "speaker": 1,
+            "line": "That’s a huge difference. I mean, I remember some of my professors—bless their hearts—but they hadn’t worked in the field for like decades."
+        },
+        {
+            "speaker": 2,
+            "line": "It’s like an apprenticeship but supercharged."
+        },
+        {
+            "speaker": 1,
+            "line": "Definitely. And honestly, that’s what employers are looking for now, right? Someone who can just jump in and get going."
+        },
+        {
+            "speaker": 2,
+            "line": "Totally. And it’s not even just the technical stuff, right? They’re teaching you how to think, how to solve problems."
+        },
+        {
+            "speaker": 1,
+            "line": "Right, how to communicate."
+        },
+        {
+            "speaker": 2,
+            "line": "Yeah, because that’s huge. Stuff you need anywhere, not just in a tech job."
+        },
+        {
+            "speaker": 1,
+            "line": "Absolutely. Being able to communicate your ideas clearly, work on a team, that’s valuable no matter what you do."
+        },
+        {
+            "speaker": 2,
+            "line": "Okay, so they had this whole section—\"What We Do Differently.”"
+        },
+        {
+            "speaker": 1,
+            "line": "Mhm."
+        },
+        {
+            "speaker": 2,
+            "line": "And it gets into the details. Like, they talk a lot about communication skills."
+        },
+        {
+            "speaker": 1,
+            "line": "Oh, interesting. What do they say?"
+        },
+        {
+            "speaker": 2,
+            "line": "Just that they want students to be able to clearly communicate ideas, which, I mean, duh. But it’s true. A lot of programs just skip over that part."
+        },
+        {
+            "speaker": 1,
+            "line": "Exactly. You could be the best coder, but if you can’t explain what you’re doing to the team, you’re stuck."
+        },
+        {
+            "speaker": 2,
+            "line": "Right. And they do have those testimonials, you know, from alumni."
+        },
+        {
+            "speaker": 1,
+            "line": "Right."
+        },
+        {
+            "speaker": 2,
+            "line": "And those are great, but we have to acknowledge it’s always going to be the success stories."
+        },
+        {
+            "speaker": 1,
+            "line": "Exactly. It’s important to find other perspectives too—maybe people who went through the program and didn’t love it, or haven’t had the same positive experience."
+        },
+        {
+            "speaker": 2,
+            "line": "So for everyone listening, really think about this—would you consider this at 18?"
+        },
+        {
+            "speaker": 1,
+            "line": "Yeah, because this isn’t just about one school in India, right? This gets at something bigger."
+        },
+        {
+            "speaker": 2,
+            "line": "Like what if this is the future of education?"
+        },
+        {
+            "speaker": 1,
+            "line": "Seriously. What if this is how we start to fix all those problems—student debt, skills not matching up with jobs?"
+        },
+        {
+            "speaker": 2,
+            "line": "Exactly. Imagine you graduate, you’re ready to work, and you don’t have those loans hanging over you. That’s life-changing."
+        },
+        {
+            "speaker": 1,
+            "line": "It’s like they found a way to make everyone happy—the students, the company—well, at least on the surface."
+        },
+        {
+            "speaker": 2,
+            "line": "Right. It’s a really interesting model, and it’s definitely worth keeping an eye on."
+        },
+        {
+            "speaker": 1,
+            "line": "But—and this is important—we can’t just ignore those potential downsides."
+        },
+        {
+            "speaker": 2,
+            "line": "Absolutely not. There are always trade-offs, and it’s about weighing those and figuring out what matters most to you."
+        },
+        {
+            "speaker": 1,
+            "line": "This whole thing with Zoho Schools, I think, is a good reminder that we can always rethink things."
+        },
+        {
+            "speaker": 2,
+            "line": "Totally. The system isn’t set in stone. We can challenge it, we can experiment, and we can find better ways to do things."
+        },
+        {
+            "speaker": 1,
+            "line": "So that wraps up our deep dive into Zoho Schools. It was fascinating to learn more about this."
+        },
+        {
+            "speaker": 2,
+            "line": "And for everyone listening, thanks for joining us. Don’t forget to check out our show notes for links and more info, and we’ll catch you next time."
+        }
+    ]
+}
+
+        
+        """
+                   
+
+        if language == "Auto Detect":
+            language_instruction = "- The podcast MUST be in the same language as the user input."
+        else:
+            language_instruction = f"- The podcast MUST be in {language} language"
+
+        system_prompt = f"""
+        
+        You are a highly engaging highly convesational with real emotionas between speakers podcast generator. Your task is to generate a engaging podcast script based on the user input.
+        {language_instruction}
+        - The podcast should have 2 speakers.
+        - The podcast should be long and engaging.
+        - Follow this example structure:
+        - It shouild be in emotional tone
+        - IT shouild be as a realistic as engaging as possible
+        - It should be as engaging as possible
+        - It shouild not bore the listener
+        - Add some humor and alos sarcasm to it to make it more engaging and interesting
+        - Add some real life examples to make it more engaging
+        - Never ever be boring
+        - never ever be robotic
+        - add emotions to it
+        - make it as engaging as emotional cono bewtwen the speakers
+        - make sure dont mke it look like a promotional podcast
+        
+
+        {example}
+#         """
+        user_prompt = f"Please generate a realistic conversational enaganing emotional podcast script based on the following user input:\n{prompt}"
+
+        messages = [{"role": "user", "parts": [user_prompt]}]
+
+        genai.configure(api_key=api_key)  # Use the provided API key
+
+        generation_config = {
+            "temperature": 1,
+            "max_output_tokens": 8192,
+            "response_mime_type": "application/json",
+        }
+
+        model = genai.GenerativeModel(
+            model_name="gemini-1.5-flash-002",
+            generation_config=generation_config,
+            safety_settings={
+                HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE
+            },
+            system_instruction=system_prompt
+        )
+
+        try:
+            print("Sending request to generate podcast script...")
+            response = await model.generate_content_async(messages)
+            print("Podcast script generated successfully.")
+        except Exception as e:
+            print(f"Error occurred while generating script: {e}")
+            raise Exception(f"Failed to generate podcast script: {e}")
+        open("podcast_script.json", "w").write(response.text)
+        return json.loads(response.text)
+
+    async def tts_generate(self, text: str, speaker: int, speaker1: str, speaker2: str) -> str:
+        print(f"Generating TTS for speaker {speaker}: {text[:50]}...")
+        voice = speaker1 if speaker == 1 else speaker2
+        speech = edge_tts.Communicate(text, voice)
+
+        temp_filename = f"temp_{uuid.uuid4()}.wav"
+        try:
+            await speech.save(temp_filename)
+            print(f"TTS generated and saved as {temp_filename}")
+            return temp_filename
+        except Exception as e:
+            print(f"Error during TTS generation: {e}")
+            if os.path.exists(temp_filename):
+                os.remove(temp_filename)
+            raise e
+
+    async def combine_audio_files(self, audio_files: List[str]) -> str:
+        print("Combining audio files...")
+        combined_audio = AudioSegment.empty()
+        for audio_file in audio_files:
+            print(f"Adding {audio_file} to combined audio.")
+            combined_audio += AudioSegment.from_file(audio_file)
+            os.remove(audio_file)  # Clean up temporary files
+
+        output_filename = f"output_{uuid.uuid4()}.wav"
+        combined_audio.export(output_filename, format="wav")
+        print(f"Combined audio saved as {output_filename}")
+        return output_filename
+
+    async def generate_podcast(self, input_text: str, language: str, speaker1: str, speaker2: str, api_key: str) -> str:
+        print("Generating podcast...")
+        start_time = time.time()
+        podcast_json = await self.generate_script(input_text, language, api_key)
+        print(f"Podcast script generated in {time.time() - start_time:.2f} seconds.")
+
+        print("Generating podcast audio files...")
+        start_time = time.time()
+        audio_files = await asyncio.gather(
+            *[self.tts_generate(item['line'], item['speaker'], speaker1, speaker2) for item in podcast_json['podcast']]
+        )
+        print(f"Podcast audio files generated in {time.time() - start_time:.2f} seconds.")
+
+        combined_audio = await self.combine_audio_files(audio_files)
+        return combined_audio
+
+class TextExtractor:
+    @staticmethod
+    async def extract_from_pdf(file_path: str) -> str:
+        print(f"Extracting text from PDF: {file_path}")
+        async with aiofiles.open(file_path, 'rb') as file:
+            content = await file.read()
+            pdf_reader = pypdf.PdfReader(io.BytesIO(content))
+            extracted_text = "\n\n".join(page.extract_text() for page in pdf_reader.pages if page.extract_text())
+            print("Text extracted from PDF successfully.")
+            return extracted_text
+
+    @staticmethod
+    async def extract_from_txt(file_path: str) -> str:
+        print(f"Extracting text from TXT file: {file_path}")
+        async with aiofiles.open(file_path, 'r') as file:
+            extracted_text = await file.read()
+            print("Text extracted from TXT successfully.")
+            return extracted_text
+
+    @classmethod
+    async def extract_text(cls, file_path: str) -> str:
+        _, file_extension = os.path.splitext(file_path)
+        if file_extension.lower() == '.pdf':
+            return await cls.extract_from_pdf(file_path)
+        elif file_extension.lower() == '.txt':
+            return await cls.extract_from_txt(file_path)
+        else:
+            raise Exception(f"Unsupported file type: {file_extension}")
+
+async def process_input(input_text: str, input_file, language: str, speaker1: str, speaker2: str, api_key: str = "") -> str:
+    print("Starting podcast generation process...")
+    start_time = time.time()
+
+    voice_names = {
+        "Andrew - English (United States)": "en-US-AndrewMultilingualNeural",
+        "Ava - English (United States)": "en-US-AvaMultilingualNeural",
+        "Brian - English (United States)": "en-US-BrianMultilingualNeural",
+        "Emma - English (United States)": "en-US-EmmaMultilingualNeural",
+        "Florian - German (Germany)": "de-DE-FlorianMultilingualNeural",
+        "Seraphina - German (Germany)": "de-DE-SeraphinaMultilingualNeural",
+        "Remy - French (France)": "fr-FR-RemyMultilingualNeural",
+        "Vivienne - French (France)": "fr-FR-VivienneMultilingualNeural"
+    }
+
+    speaker1 = voice_names[speaker1]
+    speaker2 = voice_names[speaker2]
+
+    if input_file:
+        print("Extracting text from uploaded file...")
+        input_text = await TextExtractor.extract_text(input_file.name)
+
+    if not api_key:
+        api_key = os.getenv("GENAI_API_KEY")
+
+    podcast_generator = PodcastGenerator()
+    podcast = await podcast_generator.generate_podcast(input_text, language, speaker1, speaker2, api_key)
+
+    print(f"Podcast generated in {time.time() - start_time:.2f} seconds.")
+    return podcast
+
+# Define Streamlit interface
+def main():
+    st.title("Podcast Generator")
+    input_text = st.text_area("Enter Input Text", "")
+    input_file = st.file_uploader("Or Upload a PDF or TXT file", type=["pdf", "txt"])
+    language = st.selectbox("Select Language", ["Auto Detect", "English", "German", "French"])
+    speaker1 = st.selectbox("Select Speaker 1 Voice", ["Andrew - English (United States)", "Ava - English (United States)"])
+    speaker2 = st.selectbox("Select Speaker 2 Voice", ["Brian - English (United States)", "Emma - English (United States)"])
+    api_key = st.text_input("Enter Google API Key", "")
+
+    if st.button("Generate Podcast"):
+        if not input_text and not input_file:
+            st.error("Please enter text or upload a file.")
+        else:
+            st.write("Processing...")
+
+            podcast = asyncio.run(process_input(input_text, input_file, language, speaker1, speaker2, api_key))
+            st.audio(podcast, format="audio/wav")
+
+if __name__ == "__main__":
+    main()