Update app.py

app.py

@@ -37,7 +37,7 @@ def read_pdf(file_path):
         text += pdf_reader.pages[page].extract_text()
     return text
 
-def summarize(file):
+def summarize(file, summary_length):
     # Leer el contenido del archivo subido
     file_path = file.name
     if file_path.endswith('.pdf'):
@@ -45,13 +45,20 @@ def summarize(file):
     else:
         with open(file_path, 'r', encoding='utf-8') as f:
             text = f.read()
-            
-    template = '''
+    
+    if summary_length == 'Corto':
+        length_instruction = "El resumen debe tener un máximo de 100 palabras."
+    elif summary_length == 'Medio':
+        length_instruction = "El resumen debe tener un máximo de 500 palabras."
+    else:
+        length_instruction = "El resumen debe tener un máximo de 1000 palabras."
+    
+    template = f'''
     Por favor, lea detenidamente el siguiente documento:
 <document>
-{TEXT}
+{{TEXT}}
 </document>
-Después de leer el documento, identifique los puntos clave y las ideas principales cubiertas en el texto. Organice estos puntos clave en una lista con viñetas concisa que resuma la información esencial del documento. El resumen debe tener un máximo de 10 puntos.
+Después de leer el documento, identifique los puntos clave y las ideas principales cubiertas en el texto. Organice estos puntos clave en una lista con viñetas concisa que resuma la información esencial del documento. {length_instruction}
 Su objetivo es ser exhaustivo en la captura del contenido central del documento, mientras que también es conciso en la expresión de cada punto del resumen. Omita los detalles menores y concéntrese en los temas centrales y hechos importantes.
     '''
     
@@ -102,9 +109,9 @@ Asegúrese de que la traducción sea precisa y conserve el significado original
     
     return f"Prompt:\n{formatted_prompt}\n\nTraducción:\n{translated_text.content}"
 
-def process_file(file, action, target_language=None):
+def process_file(file, action, target_language=None, summary_length=None):
     if action == "Resumen":
-        return summarize(file)
+        return summarize(file, summary_length)
     elif action == "Clasificar":
         file_path = file.name
         if file_path.endswith('.pdf'):
@@ -126,21 +133,37 @@ with gr.Blocks() as demo:
         with gr.Column():
             file = gr.File(label="Subir un archivo")
             action = gr.Radio(label="Seleccione una acción", choices=["Resumen", "Clasificar", "Traducir"])
+            summary_length = gr.Radio(label="Seleccione la longitud del resumen", choices=["Corto", "Medio", "Largo"], visible=False)
             target_language = gr.Dropdown(label="Seleccionar idioma de traducción", choices=["en", "fr", "de"], visible=False)
         
         with gr.Column():
             output_text = gr.Textbox(label="Resultado", lines=20)
 
-    def update_language_dropdown(action):
+    def update_ui(action):
         if action == "Traducir":
-            return gr.update(visible=True)
+            return gr.update(visible=False), gr.update(visible=True)
+        elif action == "Resumen":
+            return gr.update(visible=True), gr.update(visible=False)
         else:
-            return gr.update(visible=False)
+            return gr.update(visible(False)), gr.update(visible(False))
     
-    action.change(update_language_dropdown, inputs=action, outputs=target_language)
+    action.change(update_ui, inputs=action, outputs=[summary_length, target_language])
     
     submit_button = gr.Button("Procesar")
-    submit_button.click(process_file, inputs=[file, action, target_language], outputs=output_text)
+    submit_button.click(process_file, inputs=[file, action, target_language, summary_length], outputs=output_text)
+
+    def generate_file():
+        summary_text = output_text.value
+        filename = 'translation.txt' if action.value == 'Traducir' else 'summary.txt'
+        file_path = download_text(summary_text, filename)
+        return file_path
+
+    download_button = gr.Button("Descargar Resultado")
+    download_button.click(
+        fn=generate_file,
+        inputs=[],
+        outputs=gr.File()
+    )
 
 # Ejecutar la aplicación Gradio
 demo.launch(share=True)