Update app.py

app.py

@@ -11,7 +11,7 @@ model = WhisperModel("ivrit-ai/faster-whisper-v2-d4")
 # הגדרת pipeline לסיכום
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-def summarize_audio_or_video(file_path):
+def transcribe_and_summarize(file_path):
     try:
         # בדיקה אם הקובץ הוא וידאו והמרת וידאו לאודיו במידת הצורך
         if file_path.endswith((".mp4", ".mov", ".avi", ".mkv")):
@@ -30,10 +30,10 @@ def summarize_audio_or_video(file_path):
         if audio_file != file_path:
             os.remove(audio_file)
 
-        return summary
+        return transcript, summary
 
     except Exception as e:
-        return f"שגיאה בעיבוד הקובץ: {str(e)}"
+        return f"שגיאה בעיבוד הקובץ: {str(e)}", ""
 
 def convert_video_to_audio(video_file):
     # יצירת קובץ אודיו זמני
@@ -44,11 +44,11 @@ def convert_video_to_audio(video_file):
 
 # הגדרת ממשק Gradio
 interface = gr.Interface(
-    fn=summarize_audio_or_video,
+    fn=transcribe_and_summarize,
     inputs=gr.Audio(type="filepath"),
-    outputs="text",
-    title="ממיר אודיו/וידאו לסיכום",
-    description="העלה קובץ אודיו או וידאו של מרצה וקבל סיכום קצר של התוכן."
+    outputs=["text", "text"],  # פלט כפול: תמלול וסיכום
+    title="ממיר אודיו/וידאו לתמלול וסיכום",
+    description="העלה קובץ אודיו או וידאו של מרצה וקבל תמלול מלא וסיכום קצר של התוכן."
 )
 
 if __name__ == "__main__":