Added user information input

app.py

@@ -71,18 +71,17 @@ def get_full_context(input):
 is_first_run = True  # Flag to check if it's the first run
 
 # Here is the langchain
-def predict(history, input):
-
-    global is_first_run  # Use the global flag
+def predict(history, user_info, input):
+    global is_first_run
     if is_first_run:
         context = get_full_context(input)
         print(context)  # For debugging
-        is_first_run = False  # Set the flag to False after the first run
+        is_first_run = False
     else:
         context = ""
 
     history_langchain_format = []
-    history_langchain_format.append(SystemMessage(content=f"{ORIG_SYSTEM_MESSAGE_PROMPT}, here is the user information: {user_info_simulated}"))
+    history_langchain_format.append(SystemMessage(content=f"{ORIG_SYSTEM_MESSAGE_PROMPT}, here is the user information: {user_info}"))
     for human, ai in history:
         history_langchain_format.append(HumanMessage(content=human))
         history_langchain_format.append(AIMessage(content=ai))
@@ -136,22 +135,22 @@ def reset_flag():
     global is_first_run
     is_first_run = True
 
-# The Gradio App interface
+# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("""<h1><center>TROUBLESHOOTING Bot by CIONIC</center></h1>""")
-    gr.Markdown("""<p><center>To start a new case, press clear.</center></p>""")
+    gr.Markdown("""<p><center>To start a new case, enter your information and press clear.</center></p>""")
+    user_info_textbox = gr.Textbox(placeholder="Enter your information here", value="Name: Bob, Phone: iPhone 10, SN: 123456789")
     chatbot = gr.Chatbot()
     textbox = gr.Textbox()
-    clear_button = gr.ClearButton(components=[chatbot])
+    clear_button = gr.ClearButton(components=[chatbot, user_info_textbox])
     clear_button.click(reset_flag, None, None)
 
-    # Chain user and bot functions with `.then()`
+    # Submit user and bot functions
     textbox.submit(user, [textbox, chatbot], [textbox, chatbot], queue=False).then(
-        bot, chatbot, chatbot,
+        lambda args: bot(chatbot, user_info_textbox.value), chatbot, chatbot,
     )
-    # chatbot.like(vote, None, None)
 
-# Enable queuing
+# Enable queuing and launch the app
 demo.queue()
 demo.launch(debug=True)