fix

app.py

@@ -1,55 +1,35 @@
 import gradio as gr
 from utils import Bot
 
-# Define the initialization function for the bot
-def initialize_bot(openai_api_key, table_descriptions, text_documents):
-    global bot
-    bot = Bot(openai_api_key=openai_api_key, table_descriptions=table_descriptions, text_documents=text_documents, verbose=True)
-
-# Define the function to generate the bot's response
-def generate_response(question):
-    response = bot(question)
-    return response
-
-# Define the interface components
-openai_api_key_input = gr.inputs.Textbox(label="OpenAI API Key")
-table_descriptions_input = gr.inputs.Textbox(label="Table Descriptions")
-text_documents_input = gr.inputs.Textbox(label="Text Documents")
-
-submit_button = gr.outputs.Button(label="Submit")
-
-question_input = gr.inputs.Textbox(label="Ask a question")
-submit_question_button = gr.outputs.Button(label="Submit")
-
-# Define the interface layout
-input_components = [
-    openai_api_key_input,
-    table_descriptions_input,
-    text_documents_input,
-    submit_button,
-]
-
-output_components = [
-    question_input,
-    submit_question_button,
-]
-
-interface = gr.Interface(
-    initialize_bot,
-    input_components,
-    show_input=False,
-    outputs=output_components,
-    title="Chat with your Bot",
-)
-
-# Define the callback function for the submit button
-def submit_callback(openai_api_key, table_descriptions, text_documents):
-    initialize_bot(openai_api_key, table_descriptions, text_documents)
-
-# Define the callback function for the submit question button
-def submit_question_callback(question):
-    response = generate_response(question)
-    return response
-
-# Add the callbacks to the interface
-interface.launch(inline=False, submit_button=submit_callback, submit_question_button=submit_question_callback)
+def chatbot_interface():
+    openai_api_key = gr.inputs.Textbox(label="OpenAI API Key")
+    table_descriptions = gr.inputs.Textbox(label="Table Descriptions")
+    text_documents = gr.inputs.Textbox(label="Text Documents")
+    init_inputs = [openai_api_key, table_descriptions, text_documents]
+
+    def init_bot(openai_api_key, table_descriptions, text_documents):
+        bot = Bot(openai_api_key=openai_api_key, table_descriptions=table_descriptions, text_documents=text_documents)
+        return bot
+
+    chatbot = gr.inputs.Textbox(label="Ask a question")
+    submit_init_button = gr.Interface(
+        fn=init_bot,
+        inputs=init_inputs,
+        outputs=None,
+        title="Chat with your Bot",
+        description="Enter initialization parameters and click submit to start chatting with your bot."
+    )
+    submit_chat_button = gr.Interface(
+        fn=lambda bot, question: bot(question),
+        inputs=[gr.inputs.Hidden(), chatbot],
+        outputs=gr.outputs.Textbox(label="Bot Response"),
+        title="Chat with your Bot",
+        description="Type a question and click submit to get a response from your bot."
+    )
+
+    return submit_init_button, submit_chat_button
+
+submit_init_button, submit_chat_button = chatbot_interface()
+
+submit_init_button.launch()
+submit_chat_button.launch()