Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
-from utils import generate_script, generate_audio, truncate_text, extract_text_from_url
+from utils import generate_script, generate_audio, truncate_text, extract_text_from_url, summarize_text
 from prompts import SYSTEM_PROMPT
 from pydub import AudioSegment
 import pypdf
 import os
 import tempfile
 
+# Podcast generation function
 def generate_podcast(file, url, tone, length):
     try:
         if file and url:
@@ -31,13 +32,11 @@ def generate_podcast(file, url, tone, length):
         script = generate_script(SYSTEM_PROMPT, truncated_text, tone, length)
         
         audio_segments = []
-        transcript = ""
         try:
             for item in script.dialogue:
                 audio_file = generate_audio(item.text, item.speaker)
                 audio_segment = AudioSegment.from_mp3(audio_file)
                 audio_segments.append(audio_segment)
-                transcript += f"**{item.speaker}**: {item.text}\n\n"
                 os.remove(audio_file)  # Clean up temporary audio file
         except Exception as e:
             raise gr.Error(f"Error generating audio: {str(e)}")
@@ -48,32 +47,70 @@ def generate_podcast(file, url, tone, length):
             combined_audio.export(temp_audio.name, format="mp3")
             temp_audio_path = temp_audio.name
         
-        return temp_audio_path, transcript
+        return temp_audio_path
     
     except Exception as e:
         return None, f"An error occurred: {str(e)}"
 
+# Briefing document summarization function
+def summarize_document(file, url):
+    try:
+        if file and url:
+            return "Please provide either a PDF file or a URL, not both."
+        
+        if file:
+            if not file.name.lower().endswith('.pdf'):
+                return "Please upload a PDF file."
+            
+            pdf_reader = pypdf.PdfReader(file.name)
+            text = ""
+            for page in pdf_reader.pages:
+                text += page.extract_text()
+        elif url:
+            text = extract_text_from_url(url)
+        else:
+            return "Please provide either a PDF file or a URL."
+        
+        summary = summarize_text(text)  # Summarize the text using your utility function
+        return summary
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
 instructions = """
 # Podify Studio
-Welcome to the Podcast Generator ! This tool creates custom podcast episodes using AI-generated content.
+Welcome to the Podcast Generator and Briefing Document Summarizer! 
+Use this tool to generate custom podcast episodes or summarize documents.
 """
 
-iface = gr.Interface(
-    fn=generate_podcast,
-    inputs=[
-        gr.File(label="Upload PDF file (optional)", file_types=[".pdf"]),
-        gr.Textbox(label="OR Enter URL"),
-        gr.Radio(["humorous", "casual", "formal"], label="Select podcast tone", value="casual"),
-        gr.Radio(["Short (1-2 min)", "Medium (3-5 min)"], label="Podcast length", value="Medium (3-5 min)")
-    ],
-    outputs=[
-        gr.Audio(label="Generated Podcast"),
-        gr.Markdown(label="Transcript")
+# Gradio interface
+iface = gr.TabbedInterface(
+    [
+        gr.Interface(
+            fn=generate_podcast,
+            inputs=[
+                gr.File(label="Upload PDF file (optional)", file_types=[".pdf"]),
+                gr.Textbox(label="OR Enter URL"),
+                gr.Radio(["humorous", "casual", "formal"], label="Select podcast tone", value="casual"),
+                gr.Radio(["Short (1-2 min)", "Medium (3-5 min)"], label="Podcast length", value="Medium (3-5 min)")
+            ],
+            outputs=gr.Audio(label="Generated Podcast"),
+            title="🎙️ Amuthvani: AI Podcast",
+            description="Generate podcasts using PDF documents or URLs.",
+        ),
+        gr.Interface(
+            fn=summarize_document,
+            inputs=[
+                gr.File(label="Upload PDF file (optional)", file_types=[".pdf"]),
+                gr.Textbox(label="OR Enter URL")
+            ],
+            outputs=gr.Textbox(label="Summarized Document"),
+            title="📄 Briefing Document Summarizer",
+            description="Summarize documents using AI. Provide a PDF or URL to get a brief summary.",
+        )
     ],
-    title="🎙️ Amuthvani:AI Podcast !",
-    description=instructions,
-    allow_flagging="never",
-    theme=gr.themes.Soft()
+    title="Amuthvani: AI Studio",
+    description=instructions
 )
 
 if __name__ == "__main__":