Update app.py

app.py

@@ -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, text
+    return predicted_label
 
 def translate(file, target_language):
     # Leer el contenido del archivo subido
@@ -119,8 +119,7 @@ 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()
-        label, document_text = classify_text(text)
-        return f"Clasificación: {label}\n\nDocumento:\n{document_text}"
+        return classify_text(text)
     elif action == "Traducir":
         return translate(file, target_language)
     else:
@@ -166,13 +165,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(True))
+            return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
         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])
     
@@ -195,10 +194,5 @@ with gr.Blocks() as demo:
     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)