Update app.py
Browse files
app.py
CHANGED
@@ -47,11 +47,11 @@ def summarize(file, summary_length):
|
|
47 |
text = f.read()
|
48 |
|
49 |
if summary_length == 'Corto':
|
50 |
-
length_instruction = "El resumen debe tener un máximo de
|
51 |
elif summary_length == 'Medio':
|
52 |
-
length_instruction = "El resumen debe tener un máximo de
|
53 |
else:
|
54 |
-
length_instruction = "El resumen debe tener un máximo de
|
55 |
|
56 |
template = f'''
|
57 |
Por favor, lea detenidamente el siguiente documento:
|
@@ -125,15 +125,6 @@ def process_file(file, action, target_language=None, summary_length=None):
|
|
125 |
else:
|
126 |
return "Acción no válida"
|
127 |
|
128 |
-
def answer_question(text, question):
|
129 |
-
messages = [
|
130 |
-
{"role": "system", "content": "Eres un asistente útil."},
|
131 |
-
{"role": "user", "content": f"El documento es el siguiente:\n{text}"},
|
132 |
-
{"role": "user", "content": question}
|
133 |
-
]
|
134 |
-
response = llm_engine_hf.invoke(messages)
|
135 |
-
return response.content
|
136 |
-
|
137 |
def download_text(output_text, filename='output.txt'):
|
138 |
if output_text:
|
139 |
file_path = Path(filename)
|
@@ -160,27 +151,22 @@ with gr.Blocks() as demo:
|
|
160 |
|
161 |
with gr.Column():
|
162 |
output_text = gr.Textbox(label="Resultado", lines=20)
|
163 |
-
question = gr.Textbox(label="Hacer una pregunta al documento", lines=2, visible=False)
|
164 |
-
answer = gr.Textbox(label="Respuesta", lines=2, interactive=False, visible=False)
|
165 |
|
166 |
def update_visible_elements(action):
|
167 |
if action == "Traducir":
|
168 |
-
return gr.update(visible=True), gr.update(visible=False), gr.update(visible
|
169 |
elif action == "Resumen":
|
170 |
-
return gr.update(visible=False), gr.update(visible=True), gr.update(visible
|
171 |
elif action == "Clasificar":
|
172 |
-
return gr.update(visible
|
173 |
else:
|
174 |
-
return gr.update(visible
|
175 |
|
176 |
-
action.change(update_visible_elements, inputs=action, outputs=[target_language, summary_length,
|
177 |
|
178 |
submit_button = gr.Button("Procesar")
|
179 |
submit_button.click(process_file, inputs=[file, action, target_language, summary_length], outputs=output_text)
|
180 |
|
181 |
-
question_button = gr.Button("Hacer Pregunta")
|
182 |
-
question_button.click(answer_question, inputs=[output_text, question], outputs=answer)
|
183 |
-
|
184 |
def generate_file():
|
185 |
summary_text = output_text.value
|
186 |
filename = 'translation.txt' if action.value == 'Traducir' else 'summary.txt'
|
|
|
47 |
text = f.read()
|
48 |
|
49 |
if summary_length == 'Corto':
|
50 |
+
length_instruction = "El resumen debe tener un máximo de 100 palabras."
|
51 |
elif summary_length == 'Medio':
|
52 |
+
length_instruction = "El resumen debe tener un máximo de 500 palabras."
|
53 |
else:
|
54 |
+
length_instruction = "El resumen debe tener un máximo de 1000 palabras."
|
55 |
|
56 |
template = f'''
|
57 |
Por favor, lea detenidamente el siguiente documento:
|
|
|
125 |
else:
|
126 |
return "Acción no válida"
|
127 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
128 |
def download_text(output_text, filename='output.txt'):
|
129 |
if output_text:
|
130 |
file_path = Path(filename)
|
|
|
151 |
|
152 |
with gr.Column():
|
153 |
output_text = gr.Textbox(label="Resultado", lines=20)
|
|
|
|
|
154 |
|
155 |
def update_visible_elements(action):
|
156 |
if action == "Traducir":
|
157 |
+
return gr.update(visible=True), gr.update(visible=False), gr.update(visible(False))
|
158 |
elif action == "Resumen":
|
159 |
+
return gr.update(visible=False), gr.update(visible=True), gr.update(visible(False))
|
160 |
elif action == "Clasificar":
|
161 |
+
return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(True))
|
162 |
else:
|
163 |
+
return gr.update(visible(False)), gr.update(visible(False)), gr.update(visible(False))
|
164 |
|
165 |
+
action.change(update_visible_elements, inputs=action, outputs=[target_language, summary_length, output_text])
|
166 |
|
167 |
submit_button = gr.Button("Procesar")
|
168 |
submit_button.click(process_file, inputs=[file, action, target_language, summary_length], outputs=output_text)
|
169 |
|
|
|
|
|
|
|
170 |
def generate_file():
|
171 |
summary_text = output_text.value
|
172 |
filename = 'translation.txt' if action.value == 'Traducir' else 'summary.txt'
|