Update ui.py

ui.py

@@ -223,7 +223,6 @@ def add_arxiv_ids_to_queue(queue, arxiv_ids_str):
 
 def before_chat_begin():
     return (
-        gr.Button(interactive=False),
         gr.Button(interactive=False),
         gr.Button(interactive=False)
     )
@@ -260,16 +259,16 @@ async def chat_stream(idx, local_data, user_prompt, chat_state, ctx_num_lconv=3)
     try:
         async for result in gemini_gen_text(prompt):
             ppm.append_pong(result)
-            yield "", ppm.build_uis(), str(ppm), gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)
+            yield "", ppm.build_uis(), str(ppm), gr.update(interactive=False), gr.update(interactive=False)
 
-        yield "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
+        yield "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True)
     except Exception as e:
         print(str(e))
         gr.Warning("Gemini refused to answer further. This happens because there were some safety issues in the answer.")
-        yield "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)        
+        yield "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True)        
 
 def chat_reset(local_data, chat_state):
     ppm = chat_state["ppmanager_type"].from_json(json.dumps(local_data))
     ppm.pingpongs = []
 
-    return "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)
+    return "", ppm.build_uis(), str(ppm), gr.update(interactive=True), gr.update(interactive=True)