Update app.py

app.py

@@ -160,7 +160,7 @@ def predict(message, history):
         if new_token != '<':
             partial_message += new_token
             yield partial_message
-    return messages
+    return messages, source_text
 
 # Define the Gradio interface
 title = "Tchap"
@@ -172,11 +172,12 @@ examples = [
     ]
 ]
 
-demo = gr.Blocks()
-
+# Define the Gradio interface
 with gr.Blocks() as demo:
-    gr.ChatInterface(predict)
-    gr.HTML("""<h3 style="text-align:center">Sources</h3><p>""" + source_text + """</p>""")
+    with gr.Row():
+        chat_box = gr.ChatInterface(fn=predict, inputs=[gr.Text(), gr.Dataframe()])
+        source_display = gr.HTML()
+    chat_box.change(source_display.update)  # Use the change event to update the HTML component
 
 if __name__ == "__main__":
     demo.queue().launch()