Update UI

app.py

@@ -7,7 +7,7 @@ from rag import run_rag
 TOKEN = os.getenv("HF_TOKEN")
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta" , token=TOKEN)
 system_message ="You are a capable and freindly assistant."
-history = []
+
 no_change_btn = gr.Button()
 enable_btn = gr.Button(interactive=True)
 disable_btn = gr.Button(interactive=False)
@@ -22,14 +22,13 @@ class Int_State:
         print("State has been initialise")
     
     def save_question(self, question):
-    
         self.current_query = question
         self.current_response = ""
         self.history.append({"role": "user", "content": question})
         print("Question added ")
  
     def save_response(self,  assistant_message):
-        current_question = self.current_query
+        # current_question = self.current_query
         self.current_response = assistant_message
         self.history.append({"role": "system", "content": assistant_message})
         print("Response saved  ")
@@ -41,47 +40,41 @@ class Int_State:
 #  ================================================================================================================================
 state = Int_State()
 #  ================================================================================================================================
-# def clear_chat(state):
-#     state.history = []
-#     return ("",) + (disable_btn,) * 3
+def clear_chat(chatbot ):
+    state.history = []
+ 
+    chatbot = gr.Chatbot(
+                elem_id="chatbot",
+                label="Retrieval Augmented Generation (RAG) Chatbot",
+                height=300,
+                layout="bubble",
+                min_width=1200,
+                show_copy_button=True, 
+                show_share_button=True,
+                placeholder="Ask a question or type a message...",
+            )
+    yield (chatbot) + (enable_btn,) * 5
+    # yield ("" , chatbot) + (enable_btn,) * 5
 
 #  ================================================================================================================================
-# def enable_buttons(btn_list ,upvote=False, downvote=False, flag=False, regenerate=False, clear=False):
-#     return [gr.Button(interactive=upvote), gr.Button(interactive=downvote), gr.Button(interactive=flag), gr.Button(interactive=regenerate), gr.Button(interactive=clear)]
-
-
-# def disable_buttons(btn_list, upvote=True, downvote=True, flag=True, regenerate=False, clear=False):
-#     if upvote:
-#         btn_list[0] = disable_btn
-#     if downvote:
-#         btn_list[1] = disable_btn
-#     if flag:
-#         btn_list[2] = disable_btn
-#     if regenerate:
-#         btn_list[3] = disable_btn
-#     if clear:
-#         btn_list[4] = disable_btn
-#     return btn_list
-
-
-
-# def upvote_last_response(btn_list):
-#     # upvote the last response
-#     print("Upvoted")
-#     return disable_buttons(btn_list)
-
-# def downvote_last_response(btn_list):
-#     # downvote the last response
-#     print("Downvoted")
-#     return disable_buttons(btn_list)
-# def flag_last_response(btn_list):
-#     # flag the last response
-#     print("Flagged")
-#     return disable_buttons(btn_list)
+ 
+
+
+
+def upvote_last_response():
+    print("Upvoted")
+    return (disable_btn,) * 3 + (enable_btn,)*2
+
+def downvote_last_response():
+    print("Downvoted")
+    return (disable_btn,) * 3 + (enable_btn,)*2
+
+def flag_last_response():
+    print("Flagged")
+    return (disable_btn,) * 3 + (enable_btn,)*2
  
      
 def chat(
-    
     chatbot,
     message,
     max_tokens,
@@ -94,37 +87,37 @@ def chat(
     print("Temperature: ", temperature)
     print("Top P: ", top_p)
     question= message
+
     messages = [{"role": "system", "content": system_message}]
 
     history= state.get_history()
     state.save_question(message)
 
- 
     for val in history:
         messages.append(val)
 
     messages.append({"role": "user", "content": run_rag(message)})
-    response = ""
-    # chatbot.append((question))
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
+    response = "This is a response to the question"
+    # # chatbot.append((question))
+    # for msg in client.chat_completion(
+    #     messages,
+    #     max_tokens=max_tokens,
+    #     stream=True,
+    #     temperature=temperature,
+    #     top_p=top_p,
+    # ):
        
-        token = msg.choices[0].delta.content
-        response += str(token)
-        # chatbot.append(( response))
+    #     token = msg.choices[0].delta.content
+    #     response += str(token)
+        # chatbot.append(( response, response))
 
         # yield "" , chatbot 
     chatbot.append((question , response))
     state.save_response(response)
     # btn_list =enable_buttons(btn_list)
-    # yield "" , chatbot + (enable_btn,) * 5
+    yield ("" , chatbot) + (enable_btn,) * 5
 
-    yield "" , chatbot  
+    # yield "" , chatbot  
   
 
 #  ================================================================================================================================
@@ -215,37 +208,40 @@ with gr.Blocks(title="RAG", theme=theme, css=block_css , fill_height=True) as de
 #  ================================================================================================================================
     btn_list = [upvote_btn, downvote_btn, flag_btn, regenerate_btn, clear_btn]
 
-    # upvote_btn.click(
-    #     upvote_last_response,
-    #     [btn_list],
-    #     [btn_list]
-    # )
-
-    # downvote_btn.click(
-    #     downvote_last_response,
-    #     [btn_list],
-    #     [btn_list]
-    # )
-    # flag_btn.click(
-    #     flag_last_response,
-    #     [btn_list],
-    #     [btn_list]
-    # )
+    upvote_btn.click(
+        upvote_last_response,
+        [],
+        btn_list,
+    )
+
+    downvote_btn.click(
+        downvote_last_response,
+        [],
+        btn_list
+    )
+    flag_btn.click(
+        flag_last_response,
+        [],
+        btn_list,
+    )
     # regenerate_btn.click(
     #     chat,
-    #     [btn_list, chatbot, textbox, max_output_tokens, temperature, top_p],
-    #     [textbox, chatbot , btn_list]
-    # )
-    # clear_btn.click(
-    #     clear_chat,
-    #     [disable_btn] * 5
+    #     [ chatbot, textbox, max_output_tokens, temperature, top_p],
+    #     [textbox, chatbot] + btn_list
     # )
+    clear_btn.click(
+        clear_chat,
+        [chatbot],
+        [chatbot ] + btn_list,
+    )
+
     submit_btn.click(    
         chat ,
-        [chatbot, textbox , max_output_tokens, temperature, top_p],
-        [textbox ,chatbot]  
+        [ chatbot, textbox , max_output_tokens, temperature, top_p],
+        [textbox ,chatbot] + btn_list , 
                 )
 
  #  ================================================================================================================================
 demo.launch()
+
 #  ================================================================================================================================