Update app.py

app.py

@@ -23,16 +23,30 @@ def get_yt_links(contexts):
         yt_htmls.append(html.format(link, start))
     return yt_htmls
 
-def resp2msg(resp):
+'''def resp2msg(resp):
     req = resp['input']
     ans = resp['answer']
-    return req + '\n' + ans
+    return req + '\n' + ans'''
+
+def get_context(contexts):
+    txt_context = '''
+    Фрагмент 1: {}
+    Фрагмент 2: {}
+    Фрагмент 3: {}
+    '''.format(
+        contexts[0].page_content,
+        contexts[1].page_content,
+        contexts[2].page_content,
+    )
+    return txt_context
+    
     
 def process_input(text):
     response = retrieval_chain.invoke({"input": text})
     #bot.send_message(user_id, resp2msg(response))
     youtube_links = get_yt_links(response['context'])
-    return response['answer'], youtube_links[0], youtube_links[1], youtube_links[2]
+    context = get_context(response['context'])
+    return response['answer'], context, youtube_links[0], youtube_links[1], youtube_links[2]
 
 giga = os.getenv('GIGA')
 #token = os.getenv('BOT')
@@ -73,13 +87,14 @@ with gr.Blocks() as demo:
             text_input = gr.Textbox(label="Введите запрос")
             submit_btn = gr.Button("Отправить запрос")
             text_output = gr.Textbox(label="Ответ", interactive=False)
+            text_context = gr.Textbox(label="Контекст", interactive=False)
         
         with gr.Column():
             youtube_video1 = gr.HTML()
             youtube_video2 = gr.HTML()
             youtube_video3 = gr.HTML()
     
-    submit_btn.click(process_input, text_input, [text_output, youtube_video1, youtube_video2, youtube_video3])
+    submit_btn.click(process_input, text_input, [text_output, text_context, youtube_video1, youtube_video2, youtube_video3])
 
 
     demo.launch()