Update app.py

app.py

@@ -15,22 +15,22 @@ os.environ["OPENAI_API_KEY"] = apikey
 # Secret 
 # assit_Public_to_Compliance 
 # Confidential_HR_Director
-def get_answer(selected_knowledge_base, question):
-    if selected_knowledge_base == "Public":
-        assist = assit_public   # Lyft tool
-    elif selected_knowledge_base == "Secret":
-        assist = Secret   # Uber tool
-    elif selected_knowledge_base == "Public_to_Compliance":
-        assist = assit_Public_to_Compliance   # Uber tool
-    elif selected_knowledge_base == "HR_Director":
-        assist = Confidential_HR_Director  # Uber tool
+def get_answer(message, history, additional_input):
+    if additional_input == "Public":
+        assist = assit_public   
+    elif additional_input == "Secret":
+        assist = Secret   
+    elif additional_input == "Public_to_Compliance":
+        assist = assit_Public_to_Compliance   
+    elif additional_input == "HR_Director":
+        assist = Confidential_HR_Director  
     else:
         return "Invalid knowledge base selected."
 
     # # Use the selected tool to get the answer
     # response = agent.chat(question) #, tools=[selected_tool]
     interpreter_assistant = OpenAIAssistantRunnable(assistant_id= assist)
-    output = interpreter_assistant.invoke({"content": question})
+    output = interpreter_assistant.invoke({"content": message})
     response = output[0].content[0].text.value
     return response
 css = """
@@ -51,17 +51,15 @@ function refresh() {
     }
 }
 """
-gr.Interface(
-    fn=get_answer,
-    css=css, js = js_func, theme="soft", # base. citrus. soft glass. ocean
-    inputs=[
-        gr.Dropdown(
-            choices=["Public", "Secret", "Public_to_Compliance","HR_Director"],
-            label="Select Knowledge Base"
-        ),
-        gr.Textbox(label="Enter your question", lines=3)
-    ],
-    outputs="text",
-    title="Finance Companies Control System",
-    description="Ask questions related to the Finance Companies Control System.",
-).launch(debug = True)
+with gr.Blocks(css=css, js = js_func, theme="soft") as demo:
+    chatbot = gr.Chatbot(placeholder="<strong>Your Personal AI</strong><br>Ask Me Anything")
+    additional_input = gr.Dropdown(choices=["Public", "Secret", "Public_to_Compliance","HR_Director"], label="Select Knowledge Base")
+
+    gr.ChatInterface(
+        fn=get_answer,
+        type="messages",
+        chatbot=chatbot,
+        additional_inputs=[additional_input]  # Add the additional input to the ChatInterface
+    )
+
+demo.launch(show_error=True,debug=True)