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, source_text
+    return messages
 
 # Define the Gradio interface
 title = "Tchap"
@@ -172,12 +172,11 @@ examples = [
     ]
 ]
 
-# Define the Gradio interface
+demo = gr.Blocks()
+
 with gr.Blocks() as demo:
-    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
+    gr.ChatInterface(predict)
+    gr.HTML("""<h3 style="text-align:center">Sources</h3><p>""" + source_text + """</p>""")
 
 if __name__ == "__main__":
     demo.queue().launch()