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, summary_length):
+def summarize(file):
     # Leer el contenido del archivo subido
     file_path = file.name
     if file_path.endswith('.pdf'):
@@ -45,20 +45,13 @@ def summarize(file, summary_length):
     else:
         with open(file_path, 'r', encoding='utf-8') as f:
             text = f.read()
-    
-    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'''
+            
+    template = '''
     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. {length_instruction}
+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.
 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.
     '''
     
@@ -109,9 +102,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, summary_length=None):
+def process_file(file, action, target_language=None):
     if action == "Resumen":
-        return summarize(file, summary_length)
+        return summarize(file)
     elif action == "Clasificar":
         file_path = file.name
         if file_path.endswith('.pdf'):
@@ -125,19 +118,6 @@ def process_file(file, action, target_language=None, summary_length=None):
     else:
         return "Acción no válida"
 
-def download_text(output_text, filename='output.txt'):
-    if output_text:
-        file_path = Path(filename)
-        with open(file_path, 'w', encoding='utf-8') as f:
-            f.write(output_text)
-        return file_path
-    else:
-        return None
-
-def create_download_file(output_text, filename='output.txt'):
-    file_path = download_text(output_text, filename)
-    return str(file_path) if file_path else None
-
 # Crear la interfaz de Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## Procesador de Documentos")
@@ -147,38 +127,20 @@ with gr.Blocks() as demo:
             file = gr.File(label="Subir un archivo")
             action = gr.Radio(label="Seleccione una acción", choices=["Resumen", "Clasificar", "Traducir"])
             target_language = gr.Dropdown(label="Seleccionar idioma de traducción", choices=["en", "fr", "de"], visible=False)
-            summary_length = gr.Radio(label="Seleccione la longitud del resumen", choices=["Corto", "Medio", "Largo"], visible=False)
         
         with gr.Column():
             output_text = gr.Textbox(label="Resultado", lines=20)
 
-    def update_visible_elements(action):
+    def update_language_dropdown(action):
         if action == "Traducir":
-            return gr.update(visible=True), gr.update(visible=False), gr.update(visible(False))
-        elif action == "Resumen":
-            return gr.update(visible=False), gr.update(visible=True), gr.update(visible(False))
-        elif action == "Clasificar":
-            return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(True))
+            return gr.update(visible=True)
         else:
-            return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False))
+            return gr.update(visible=False)
     
-    action.change(update_visible_elements, inputs=action, outputs=[target_language, summary_length, output_text])
+    action.change(update_language_dropdown, inputs=action, outputs=target_language)
     
     submit_button = gr.Button("Procesar")
-    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()
-    )
+    submit_button.click(process_file, inputs=[file, action, target_language], outputs=output_text)
 
 # Ejecutar la aplicación Gradio
 demo.launch(share=True)