tried to change the ui a bit , making similar to the colab notebook

app.py

@@ -149,20 +149,20 @@ def chatbot(message, history):
     
     return total_text
 
-iface = gr.Interface(
-    fn=chatbot,
-    inputs="text",
-    outputs="text",
+iface = gr.ChatInterface(
+    chatbot,
     title="LLM Research Assistant",
     description="Ask questions about LLM architectures, datasets, and training techniques.",
     examples=[
-        ["What are some milestone model architectures in LLMs?"],
-        ["Explain the transformer architecture."],
-        ["Tell me about datasets used to train LLMs."],
-        ["How are LLM training datasets cleaned and preprocessed?"],
-        ["Summarize the user queries so far"]
+        "What are some milestone model architectures in LLMs?",
+        "Explain the transformer architecture.",
+        "Tell me about datasets used to train LLMs.",
+        "How are LLM training datasets cleaned and preprocessed?",
+        "Summarize the user queries so far"
     ],
-    allow_flagging="never"
+    retry_btn="Regenerate",
+    undo_btn="Undo",
+    clear_btn="Clear",
 )
 
 if __name__ == "__main__":