Update app.py

app.py

@@ -2,9 +2,6 @@ from response_parser import *
 import gradio as gr
 import os
 
-USERNAME = os.getenv('USERNAME')
-PASSWORD = os.getenv('PASSWORD')
-
 
 def initialization(state_dict: Dict) -> None:
     if not os.path.exists('cache'):
@@ -19,12 +16,6 @@ def get_bot_backend(state_dict: Dict) -> BotBackend:
     return state_dict["bot_backend"]
 
 
-def switch_to_gpt4(state_dict: Dict, whether_switch: bool) -> None:
-    bot_backend = get_bot_backend(state_dict)
-    if whether_switch:
-        bot_backend.update_gpt_model_choice("GPT-4")
-    else:
-        bot_backend.update_gpt_model_choice("GPT-3.5")
 
 
 def add_text(state_dict: Dict, history: List, text: str) -> Tuple[List, Dict]:
@@ -36,62 +27,6 @@ def add_text(state_dict: Dict, history: List, text: str) -> Tuple[List, Dict]:
     return history, gr.update(value="", interactive=False)
 
 
-def add_file(state_dict: Dict, history: List, file) -> List:
-    bot_backend = get_bot_backend(state_dict)
-    path = file.name
-    filename = os.path.basename(path)
-
-    bot_msg = [f'📁[{filename}]', None]
-    history.append(bot_msg)
-
-    bot_backend.add_file_message(path=path, bot_msg=bot_msg)
-
-    return history
-
-
-def undo_upload_file(state_dict: Dict, history: List) -> Tuple[List, Dict]:
-    bot_backend = get_bot_backend(state_dict)
-    bot_msg = bot_backend.revoke_file()
-
-    if bot_msg is None:
-        return history, gr.Button.update(interactive=False)
-
-    else:
-        assert history[-1] == bot_msg
-        del history[-1]
-        if bot_backend.revocable_files:
-            return history, gr.Button.update(interactive=True)
-        else:
-            return history, gr.Button.update(interactive=False)
-
-
-def refresh_file_display(state_dict: Dict) -> List[str]:
-    bot_backend = get_bot_backend(state_dict)
-    work_dir = bot_backend.jupyter_work_dir
-    filenames = os.listdir(work_dir)
-    paths = []
-    for filename in filenames:
-        paths.append(
-            os.path.join(work_dir, filename)
-        )
-    return paths
-
-
-def restart_ui(history: List) -> Tuple[List, Dict, Dict, Dict, Dict]:
-    history.clear()
-    return (
-        history,
-        gr.Textbox.update(value="", interactive=False),
-        gr.Button.update(interactive=False),
-        gr.Button.update(interactive=False),
-        gr.Button.update(interactive=False)
-    )
-
-
-def restart_bot_backend(state_dict: Dict) -> None:
-    bot_backend = get_bot_backend(state_dict)
-    bot_backend.restart()
-
 
 def bot(state_dict: Dict, history: List) -> List:
     bot_backend = get_bot_backend(state_dict)
@@ -135,14 +70,8 @@ if __name__ == '__main__':
                         placeholder="Enter text and press enter, or upload a file",
                         container=False
                     )
-                with gr.Column(scale=0.15, min_width=0):
-                    file_upload_button = gr.UploadButton("📁", file_types=['file'])
             with gr.Row(equal_height=True):
-                with gr.Column(scale=0.7):
-                    check_box = gr.Checkbox(label="Use GPT-4", interactive=config['model']['GPT-4']['available'])
-                    check_box.change(fn=switch_to_gpt4, inputs=[state, check_box])
-                with gr.Column(scale=0.15, min_width=0):
-                    restart_button = gr.Button(value='🔄 Restart')
+                
                 with gr.Column(scale=0.15, min_width=0):
                     undo_file_button = gr.Button(value="↩️Undo upload file", interactive=False)
         with gr.Tab("Files"):
@@ -152,38 +81,9 @@ if __name__ == '__main__':
         txt_msg = text_box.submit(add_text, [state, chatbot, text_box], [chatbot, text_box], queue=False).then(
             bot, [state, chatbot], chatbot
         )
-        txt_msg.then(fn=refresh_file_display, inputs=[state], outputs=[file_output])
-        txt_msg.then(lambda: gr.update(interactive=True), None, [text_box], queue=False)
-        txt_msg.then(lambda: gr.Button.update(interactive=False), None, [undo_file_button], queue=False)
-
-        file_msg = file_upload_button.upload(
-            add_file, [state, chatbot, file_upload_button], [chatbot], queue=False
-        ).then(
-            bot, [state, chatbot], chatbot
-        )
-        file_msg.then(lambda: gr.Button.update(interactive=True), None, [undo_file_button], queue=False)
-        file_msg.then(fn=refresh_file_display, inputs=[state], outputs=[file_output])
-
-        undo_file_button.click(
-            fn=undo_upload_file, inputs=[state, chatbot], outputs=[chatbot, undo_file_button]
-        ).then(
-            fn=refresh_file_display, inputs=[state], outputs=[file_output]
-        )
-
-        restart_button.click(
-            fn=restart_ui, inputs=[chatbot],
-            outputs=[chatbot, text_box, restart_button, file_upload_button, undo_file_button]
-        ).then(
-            fn=restart_bot_backend, inputs=[state], queue=False
-        ).then(
-            fn=refresh_file_display, inputs=[state], outputs=[file_output]
-        ).then(
-            fn=lambda: (gr.Textbox.update(interactive=True), gr.Button.update(interactive=True),
-                        gr.Button.update(interactive=True)),
-            inputs=None, outputs=[text_box, restart_button, file_upload_button], queue=False
-        )
+       
 
         block.load(fn=initialization, inputs=[state])
 
     block.queue()
-    block.launch(inbrowser=True , auth=(USERNAME, PASSWORD))
+    block.launch(inbrowser=True)