añadidas mejoras visuales

app.py

@@ -170,8 +170,8 @@ def search_and_answer(
     final_answer = answers[0]["generated_text"]
     if do_tts:
         audio_answer = tts_es(remove_chars_to_tts(final_answer))
-    final_answer = parse_final_answer(final_answer, contexts)
-    return final_answer, audio_answer if do_tts else "audio_troll.flac"
+    final_answer, documents = parse_final_answer(final_answer, contexts)
+    return final_answer, documents, audio_answer if do_tts else "audio_troll.flac"
 
 
 if __name__ == "__main__":
@@ -231,7 +231,10 @@ if __name__ == "__main__":
         ],
         outputs=[
             gr.outputs.HTML(
-                label="Answer from the system."
+                label="Generated Answer."
+            ),
+            gr.outputs.HTML(
+                label="Documents used."
             ),
             gr.outputs.Audio(label="Answer in audio"),
         ],

article_app.py

@@ -61,6 +61,18 @@ examples = [
         "wav2vec2-iic",
         False,
     ],
+    [
+        "¿Cómo se genera la apendicitis?",
+        "vacio.flac",
+        "vacio.flac",
+        50,
+        8,
+        3,
+        1.0,
+        250,
+        "wav2vec2-iic",
+        False,
+    ],
     [
         "¿Qué es la mesoterapia?",
         "vacio.flac",

general_utils.py

@@ -121,18 +121,17 @@ def transcript(input_file, audio_array, processor, model):
     whole_text = add_question_symbols(whole_text)
     return whole_text
 
-
+# <em>Final Answer:</em> 
 def parse_final_answer(answer_text: str, contexts: List):
     """Parse the final answer into correct format"""
-    s = (
-        f"<b><em>Final Answer:</em> {answer_text}</b> \n\n\n"
-        + "<p> Contexts Used: \n <p>"
-        + "\n".join(
+    answer = f"<p><b>{answer_text}</b></p> \n\n\n"
+    docs = (
+        "\n".join(
             [
-                ("""<p style="text-align: justify;">""" + context)[:300]
+                ("""<p style="text-align: justify;">""" + context)[:250]
                 + "[...]</p>"
                 for context in contexts[:5]
             ]
         )
     )
-    return s
+    return answer, docs