Update app.py

app.py

@@ -47,11 +47,11 @@ def summarize(file, summary_length):
             text = f.read()
     
     if summary_length == 'Corto':
-        length_instruction = "El resumen debe tener un máximo de 5 puntos."
+        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 10 puntos."
+        length_instruction = "El resumen debe tener un máximo de 500 palabras."
     else:
-        length_instruction = "El resumen debe tener un máximo de 15 puntos."
+        length_instruction = "El resumen debe tener un máximo de 1000 palabras."
     
     template = f'''
     Por favor, lea detenidamente el siguiente documento:
@@ -80,7 +80,7 @@ def classify_text(text):
     logits = outputs.logits
     predicted_class_id = logits.argmax(dim=-1).item()
     predicted_label = id2label[predicted_class_id]
-    return predicted_label
+    return predicted_label, text
 
 def translate(file, target_language):
     # Leer el contenido del archivo subido
@@ -119,7 +119,8 @@ def process_file(file, action, target_language=None, summary_length=None):
         else:
             with open(file_path, 'r', encoding='utf-8') as f:
                 text = f.read()
-        return classify_text(text)
+        label, document_text = classify_text(text)
+        return f"Clasificación: {label}\n\nDocumento:\n{document_text}"
     elif action == "Traducir":
         return translate(file, target_language)
     else:
@@ -149,7 +150,7 @@ def create_download_file(output_text, filename='output.txt'):
 
 # Crear la interfaz de Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("## Procesador de Documentos")
+    gr.Markdown("## LexAIcon")
 
     with gr.Row():
         with gr.Column():
@@ -165,13 +166,13 @@ with gr.Blocks() as demo:
 
     def update_visible_elements(action):
         if action == "Traducir":
-            return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+            return gr.update(visible=True), gr.update(visible=False), gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False))
         elif action == "Resumen":
-            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+            return gr.update(visible(False)), gr.update(visible(True)), gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False))
         elif action == "Clasificar":
-            return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
+            return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(True)), gr.update(visible(True)), gr.update(visible(True))
         else:
-            return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible(False)), gr.update(visible=False)
+            return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False))
     
     action.change(update_visible_elements, inputs=action, outputs=[target_language, summary_length, question, output_text, answer])
     
@@ -187,12 +188,17 @@ with gr.Blocks() as demo:
     download_button = gr.Button("Descargar Resultado")
     download_button.click(
         fn=generate_file,
-        inputs=[],
+        inputs=[output_text],
         outputs=gr.File()
     )
 
     question_button = gr.Button("Hacer Pregunta")
     question_button.click(answer_question, inputs=[output_text, question], outputs=answer)
 
+    with gr.Row():
+        question.render()
+        question_button.render()
+        answer.render()
+
 # Ejecutar la aplicación Gradio
-demo.launch(share=True)
+demo.launch(share=True)