Update some code styles

app.py

@@ -57,7 +57,7 @@ class MediatorRetriever(BaseRetriever):
                 }
             }
         ]
-
+        
         response = openai_client.chat.completions.create(
             model="gpt-4-1106-preview",
             messages=[
@@ -162,11 +162,10 @@ class MediatorRetriever(BaseRetriever):
                     response = openai_client.chat.completions.create(
                         model="gpt-4-1106-preview",
                         messages=[
-                            {"role": "system", "content": "Please extract how many mediators users want to search."},
+                            {"role": "system", "content": "Please extract how many mediators users want to search for."},
                             {"role": "user", "content": query}
                         ],
-                        functions=tools,
-                        stream=True
+                        functions=tools
                     )
                     try:
                         number_str = response.choices[0].message.function_call.arguments
@@ -200,7 +199,7 @@ class MediatorRetriever(BaseRetriever):
                         filter=metadata,
                         include_metadata=True
                     )
-                    
+
                     new_data = []
                     for result in results['matches']:
                         data = {}
@@ -400,7 +399,7 @@ js_code = """
 
     
     function autoScroll() {
-        const chatbox = document.querySelector('#component-4');
+        const chatbox = document.querySelector('.bubble-wrap');
         if (chatbox) {
             chatbox.scrollTop = chatbox.scrollHeight;
         }
@@ -414,11 +413,19 @@ js_code = """
 </script>
 """
 
+def toggle_chatbox():
+    global chatbox_visible
+    chatbox_visible = not chatbox_visible
+    return gr.update(visible=chatbox_visible)
+
+chatbox_visible = True
+
 with gr.Blocks(head=js_code) as demo:
-    with gr.Column() as chatbox_container:
+    with gr.Column(visible=chatbox_visible) as chatbox_container:
         chatbox = gr.ChatInterface(fn=search, title="Mediate.com Chatbot Application", multimodal=False, retry_btn=None, clear_btn=None, undo_btn=None, chatbot=chatbot)
     
-    gr.HTML(f'<button id="message-btn"/><img style="display: inline;" src="/file=message.png" id="custom-btn"/><button>')
+ 
+    gr.HTML(f'<button src="/file=message.png" id="message-btn"/><img style="display: inline;" src="/file=message.png" id="custom-btn"/><button>')
 
 if __name__ == "__main__":
     demo.launch(debug=True, allowed_paths=["./"])