Update app.py

app.py

@@ -12,7 +12,7 @@ enable_btn = gr.Button.update(interactive=True)
 import sqlite3
 import requests
 
-def classify_vote(user_input):
+async def classify_vote(user_input):
     url = "https://api-inference.huggingface.co/models/facebook/bart-large-mnli"
     headers = {
         "accept": "*/*",
@@ -26,14 +26,16 @@ def classify_vote(user_input):
             "multi_class": True
         }
     }
-    response = requests.post(url, headers=headers, json=payload)
-    if response.status_code == 200:
-        response_data = response.json()
-        top_category = response_data["labels"][0].strip()
-        return top_category
-    else:
-        print(f"Error: {response.status_code}")
-        return None
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, json=payload) as response:
+            if response.status == 200:
+                response_data = await response.json()
+                top_category = response_data["labels"][0].strip()
+                return top_category
+            else:
+                print(f"Error: {response.status}")
+                return None
+                
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 async def direct_regenerate(model, user_input, chatbot, character_name, character_description, user_name):
@@ -168,24 +170,40 @@ def update_ratings(state, winner_index, collection, category):
     except Exception as e:
         print(f"Error updating ratings: {str(e)}")
         return [('Error', 'Failed to update ratings.'), ('Error', 'Failed to update ratings.')]
-def vote_up_model(state, chatbot, chatbot2, character_name, character_description, user_name):
+async def vote_up_model(state, chatbot, chatbot2, character_name, character_description, user_name):
     user_input = format_prompt(state, 0, character_name, character_description, user_name)
     collection = init_database()
-    top_category = classify_vote(user_input)
+    
+    # Disable both upvote buttons immediately
+    yield chatbot, chatbot2, gr.Button.update(interactive=False, value="👍 Upvoted"), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
+    
+    # Process sentiment analysis asynchronously
+    sentiment_task = asyncio.create_task(classify_vote(user_input))
+    top_category = await sentiment_task
+    
     if top_category:
         update_message = update_ratings(state, 0, collection, top_category)
         chatbot.append(update_message[0])
         chatbot2.append(update_message[1])
+    
     return chatbot, chatbot2, gr.Button.update(interactive=False, value="👍 Upvoted"), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
-
-def vote_down_model(state, chatbot, chatbot2, character_name, character_description, user_name):
+    
+async def vote_down_model(state, chatbot, chatbot2, character_name, character_description, user_name):
     user_input = format_prompt(state, 1, character_name, character_description, user_name)
     collection = init_database()
-    top_category = classify_vote(user_input)
+    
+    # Disable both upvote buttons immediately
+    yield chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False, value="👍 Upvoted"), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
+    
+    # Process sentiment analysis asynchronously
+    sentiment_task = asyncio.create_task(classify_vote(user_input))
+    top_category = await sentiment_task
+    
     if top_category:
         update_message = update_ratings(state, 1, collection, top_category)
         chatbot2.append(update_message[0])
         chatbot.append(update_message[1])
+    
     return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False, value="👍 Upvoted"), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
     
 async def user_ask(state, chatbot1, chatbot2, textbox, character_name, character_description, user_name):
@@ -367,9 +385,8 @@ with gr.Blocks() as demo:
         textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox, character_name, character_description, user_name], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
         collection = init_database()
 
-        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-
+        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn], api_name="upvote_a")
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, character_name, character_description, user_name], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn], api_name="upvote_b")
     # ...
     with gr.Tab("💬 Direct Chat"):
         gr.Markdown("## 🗣️ Chat directly with a model!")