add highlight

app.py

@@ -28,7 +28,8 @@ def get_answer(user_input, decoding_methods, num_beams, top_k, top_p, temperatur
                                      penalty_alpha=penalty_alpha)
     answer = generated_text[0]["generated_text"]
     answer_without_prompt = answer[len(prompt)+1:]
-    return answer_without_prompt
+
+    return [(f"{user_input} ", None), (answer_without_prompt, "")]
 
 
 with gr.Blocks() as demo:
@@ -53,7 +54,11 @@ with gr.Blocks() as demo:
             with gr.Row():
                 button_generate_story = gr.Button("Submit")
         with gr.Column():
-            generated_answer = gr.Textbox()
+            # generated_answer = gr.Textbox()
+            generated_answer = gr.HighlightedText(
+                label="Answer",
+                combine_adjacent=True,
+            ).style(color_map={"": "blue", "-": "green"})
     with gr.Row():
         gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=cahya_indochat)")