Add Gradio app for audio transcription and summarization

app.py

@@ -1,33 +1,35 @@
-import whisper
 import gradio as gr
+import whisper
 from transformers import pipeline
 
 # Load models
-def transcribe_audio(file_path):
-    model = whisper.load_model("tiny")  # Use "tiny", "base", "small", etc.
-    result = model.transcribe(file_path)
-    return result["text"]
+model = whisper.load_model("base")
+summarizer = pipeline("summarization", model="t5-small")
 
-def extract_topics(text):
-    summarizer = pipeline("summarization")
-    summary = summarizer(text, max_length=50, min_length=25, do_sample=False)
-    return summary[0]["summary_text"]
+# Function to transcribe and summarize
+def transcribe_and_summarize(audio_file):
+    # Transcription
+    result = model.transcribe(audio_file)
+    transcription = result["text"]
+    
+    # Summarization
+    summary = summarizer(transcription, max_length=50, min_length=10, do_sample=False)[0]["summary_text"]
+    return transcription, summary
 
-def process_audio(file):
-    # Transcribe the audio file
-    transcript = transcribe_audio(file.name)
-    # Extract topics from the transcription
-    topics = extract_topics(transcript)
-    return transcript, topics
+# Gradio Interface
+inputs = gr.Audio(type="filepath", label="Upload your audio file")
+outputs = [
+    gr.Textbox(label="Transcription"),
+    gr.Textbox(label="Summary")
+]
 
-# Gradio interface
-interface = gr.Interface(
-    fn=process_audio,
-    inputs=gr.Audio(source="upload", type="filepath"),
-    outputs=["text", "text"],
-    title="Audio Transcription and Topic Extraction",
-    description="Upload an audio file to get a transcription and extract main topics."
+app = gr.Interface(
+    fn=transcribe_and_summarize,
+    inputs=inputs,
+    outputs=outputs,
+    title="Audio Transcription and Summarization",
+    description="Upload an audio file to get its transcription and a summarized version of the content."
 )
 
-if __name__ == "__main__":
-    interface.launch()
+# Launch the app
+app.launch()