import gradio as gr
from transformers import pipeline

# יצירת pipe לתמלול ולסיכום
transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base")
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

def summarize_audio(audio_file):
    # תמלול האודיו
    transcript = transcriber(audio_file)["text"]
    # יצירת סיכום של התמלול
    summary = summarizer(transcript, max_length=50, min_length=25, do_sample=False)[0]["summary_text"]
    return summary

# הגדרת ממשק Gradio
interface = gr.Interface(
    fn=summarize_audio,
    inputs=gr.Audio(source="upload", type="filepath"),
    outputs="text",
    title="ממיר אודיו לסיכום",
    description="העלה קובץ אודיו של מרצה וקבל סיכום קצר של התוכן."
)

if __name__ == "__main__":
    interface.launch()