Update app.py

app.py

@@ -1,77 +1,73 @@
-from IPython import get_ipython
-from IPython.display import display
-# %%
-import gradio as gr
-from transformers import pipeline
-import torch
-from gtts import gTTS
-import os
-from PyPDF2 import PdfReader
-
-# --- Dispositivo ---
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
- 
-# --- Summarizer en español (modelo multilingüe compatible) ---
-summarizer = pipeline(
-    "summarization",
-    model="csebuetnlp/mT5_multilingual_XLSum",
-    tokenizer="csebuetnlp/mT5_multilingual_XLSum",
-    device=0 if torch.cuda.is_available() else -1
-)
-
-# ... (other imports and functions remain the same)
- 
-def summarize_and_speak(input_type, text_input, pdf_input):
-    """
-    Summarizes the input data (text or PDF) and converts the summary to speech.
-    """
-    try:
-        if input_type == "text":
-            text = text_input
-        elif input_type == "pdf":
-            reader = PdfReader(pdf_input.name)  # Assuming input_data is a file-like object
-            text = ""
-            for page in reader.pages:
-                text += page.extract_text()
-        else:
-            raise ValueError("Invalid input type. Choose 'text' or 'pdf'.")
- 
-        # --- Usando el nuevo modelo de summarize ---
-        summary = summarizer(
-            text,
-            max_length=300,
-            min_length=80,
-            do_sample=False
-        )[0]["summary_text"]
- 
-        tts = gTTS(text=summary, lang='es')
-        tts.save("summary.mp3")
-        return summary, "summary.mp3"
-    except Exception as e:
-        return f"An error occurred: {e}", None
- 
-with gr.Blocks() as demo:
-    gr.Markdown("## Resumen de Historias con Voz")
-   
-    with gr.Tab("Texto"):
-        text_input = gr.Textbox(label="Introduce tu historia aquí")
-    with gr.Tab("PDF"):
-        pdf_input = gr.File(label="Sube tu archivo PDF", type="filepath")
-   
-    with gr.Row():
-        text_output = gr.Textbox(label="Resumen")
-        audio_output = gr.Audio(label="Resumen en Audio")
- 
-    submit_btn = gr.Button("Resumir y Convertir a Voz")
-   
-    # Updated to use gr.components for input elements
-    # The 'default' keyword argument is replaced by setting the value directly.
-    input_type = gr.components.Radio(choices=["text", "pdf"], label="Tipo de entrada")
-    input_type.value = "text"  # Set the default value here
- 
-    # Pass all inputs to the function, and the function will decide which one to use
-    submit_btn.click(fn=summarize_and_speak,
-                     inputs=[input_type, text_input, pdf_input],
-                     outputs=[text_output, audio_output])
- 
+import gradio as gr
+from transformers import pipeline
+import torch
+from gtts import gTTS
+from PyPDF2 import PdfReader
+
+# --- Dispositivo ---
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+ 
+# --- Summarizer en español (modelo multilingüe compatible) ---
+summarizer = pipeline(
+    "summarization",
+    model="csebuetnlp/mT5_multilingual_XLSum",
+    tokenizer="csebuetnlp/mT5_multilingual_XLSum",
+    device=0 if torch.cuda.is_available() else -1
+)
+
+# ... (other imports and functions remain the same)
+ 
+def summarize_and_speak(input_type, text_input, pdf_input):
+    """
+    Summarizes the input data (text or PDF) and converts the summary to speech.
+    """
+    try:
+        if input_type == "text":
+            text = text_input
+        elif input_type == "pdf":
+            reader = PdfReader(pdf_input.name)  # Assuming input_data is a file-like object
+            text = ""
+            for page in reader.pages:
+                text += page.extract_text()
+        else:
+            raise ValueError("Invalid input type. Choose 'text' or 'pdf'.")
+ 
+        # --- Usando el nuevo modelo de summarize ---
+        summary = summarizer(
+            text,
+            max_length=300,
+            min_length=80,
+            do_sample=False
+        )[0]["summary_text"]
+ 
+        tts = gTTS(text=summary, lang='es')
+        tts.save("summary.mp3")
+        return summary, "summary.mp3"
+    except Exception as e:
+        return f"An error occurred: {e}", None
+ 
+with gr.Blocks() as demo:
+    gr.Markdown("## Resumen de Historias con Voz")
+   
+    with gr.Tab("Texto"):
+        text_input = gr.Textbox(label="Introduce tu historia aquí")
+    with gr.Tab("PDF"):
+        pdf_input = gr.File(label="Sube tu archivo PDF", type="filepath")
+   
+    with gr.Row():
+        text_output = gr.Textbox(label="Resumen")
+        audio_output = gr.Audio(label="Resumen en Audio")
+ 
+    submit_btn = gr.Button("Resumir y Convertir a Voz")
+   
+    # Updated to use gr.components for input elements
+    # The 'default' keyword argument is replaced by setting the value directly.
+    input_type = gr.components.Radio(choices=["text", "pdf"], label="Tipo de entrada")
+    input_type.value = "text"  # Set the default value here
+ 
+    # Pass all inputs to the function, and the function will decide which one to use
+    submit_btn.click(fn=summarize_and_speak,
+                     inputs=[input_type, text_input, pdf_input],
+                     outputs=[text_output, audio_output])
+ 
 demo.launch(share=True)