Update app.py

app.py

@@ -82,7 +82,50 @@ def load_demo_refresh_model_list(request: gr.Request):
     return state, dropdown_update
 
 
-def vote_last_response(state, vote_type, model_selector, request: gr.Request):
+def vote_dummy_fn(tmp, index_state, data: gr.LikeData):
+    value_new = data.value
+    index_new = data.index
+    if len(index_state) == 0 :
+        index_state.append(index_new)
+    else:
+        if index_new in index_state:
+            gr.Warning('Your feedback is already saved.')
+            return index_state
+        else:
+            index_state.append(index_new)
+
+    #return str(data.value) + ";" + str(data.index)+";"+ str(data.liked)+";"+str(index_state), index_state
+    return index_state 
+    
+#chatbot.like(vote_last_response, [model_selector, index_state], [index_state])
+def vote_last_response(chatbot, model_selector, index_state, request: gr.Request, data: gr.LikeData):
+
+    vote_type = "upvote" if data.liked else "downvote"
+    vote_value = data.value
+    vote_index = data.index
+    if len(index_state) == 0 :
+        index_state.append(vote_index)
+    else:
+        if vote_index in index_state:
+            gr.Warning('Your feedback is already saved.')
+            return index_state
+        else:
+            index_state.append(index_new)
+            
+    with open(get_conv_log_filename(), "a") as fout:
+        data = {
+            "tstamp": round(time.time(), 4),
+            "type": vote_type,
+            "model": model_selector,
+            "state": chatbot[vote_index[0]],
+            "ip": request.client.host,
+        }
+        logger.info(f"^^^^^ data is - {data}")
+        fout.write(json.dumps(data) + "\n")
+        
+    return index_state
+
+def vote_last_response_old(state, vote_type, model_selector, request: gr.Request):
     with open(get_conv_log_filename(), "a") as fout:
         data = {
             "tstamp": round(time.time(), 4),
@@ -504,6 +547,11 @@ def build_demo(embed_mode):
             [state, chatbot] + btn_list,
         )
 
+        # For adding voting option to chatbot
+        chatbot.like(vote_last_response, [chatbot, model_selector, index_state], [index_state])
+        #def vote_last_response(chatbot, model_selector, index_state, request: gr.Request, data: gr.LikeData):
+        #state, model_selector, request: gr.Request
+        
         if args.model_list_mode == "once":
             demo.load(
                 load_demo,