Update app.py

app.py

@@ -32,6 +32,7 @@ user_db = {os.environ['username1']: os.environ['password1'], os.environ['usernam
 
 messages = [{"role": "system", "content": 'You are a helpful assistant.'}]
 errors = []
+error_recorded = 1
 
 
 
@@ -95,19 +96,44 @@ def delete_pinecone(index_name):
 # Record feed back
 
 def not_in_error():
-    return
+    global error_recorded
+    if(error_recorded):
+        return
+    else:
+        global messages
+        error_recorded = 1
+    error = ("not_in_error", messages)
+    errors.append(error)
+    return "Thank you, the question has been marked as not in ROCm."
 
 
 def not_found_error():
-    return stats
-
-
-
-def llm_error ():
-    return
+    global error_recorded
+    if(error_recorded):
+        return
+    else:
+        global messages
+        error_recorded = 1
+    error = ("not_found_error", messages)
+    errors.append(error)
+    return "Thank you, the context has been marked as not found in Vector Server."
+
+
+
+def llm_error():
+    global error_recorded
+    if(error_recorded):
+        return
+    else:
+        global messages
+        error_recorded = 1
+    error = ("not_found_error", messages)
+    errors.append(error)
+    return "Thank you, the LLM error has been recorded correctly."
 
 def list_erros():
-    return
+    global errors
+    return errors
 
 
 
@@ -129,7 +155,9 @@ def roleChoice(role):
 
 def talk2file(index_name, text):
     #disable the global message
-    #global messages
+    global messages
+    global error_recorded
+    error_recorded = 0
     messages = [{"role": "system", "content": 'You are a helpful assistant.'}]
     
     #same as filesearch
@@ -199,7 +227,14 @@ def show():
 # feed back record
 with gr.Blocks() as FeedBack:
     gr.Markdown("Record Feedback for the ROCm Usage Tutor, using the following three buttons to record three different errors:\n 1. ROCm-related contexts not included in the ROCm repo.\n 2. Context included in the repo, but the vector server fails to find it.\n 3. The LLM model fails to understand the context.")
-
+    notin_btn = gr.Button("Not_In_Err")
+    notin_btn.click(fn=not_in_error, inputs=None, outputs="text", queue=False)
+    notfound_btn = gr.Button("Not_Found_Err")
+    notfound_btn.click(fn=not_found_error, inputs=None, outputs="text", queue=False)
+    llmerr_btn = gr.Button("LLM_Err")
+    llmerr_btn.click(fn=llm_error, inputs=None, outputs="text", queue=False)
+    listerr_btn = gr.Button("List_Erros")
+    listerr_btn.click(fn=list_erros, inputs=None, outputs="text", queue=False)