some small changes

app.py

@@ -40,9 +40,9 @@ tree_depth = parameters_section.number_input("Tree Depth", min_value=1, max_valu
 iterations = parameters_section.number_input("Iterations", min_value=1, max_value=4, value=2)
 sidebar.markdown('<hr style="margin-top: 0.5rem; margin-bottom: 0.5rem;">', unsafe_allow_html=True)
 
-with sidebar:
-    runtime_container = st.container()
-    runtime_container.empty()
+# with sidebar:
+#     runtime_container = st.container()
+#     runtime_container.empty()
 
 runtime_messages = []
 
@@ -81,9 +81,9 @@ def run_querry():
                 response = lats_main(args)
 
         sys.stdout = old_stdout
-        # runtime_container.write("Response fetched.")
-        # chat_col.markdown('<hr style="margin-top: 0.5rem; margin-bottom: 0.5rem;">', unsafe_allow_html=True)
-        # chat_col.write(f"```python\n{response} \n")
+        runtime_container.write("Response fetched.")
+        chat_col.markdown('<hr style="margin-top: 0.5rem; margin-bottom: 0.5rem;">', unsafe_allow_html=True)
+        chat_col.write(f"```python\n{response} \n")
 
         return response