Update app.py

app.py

@@ -2,7 +2,6 @@ import google.generativeai as palm
 import gradio as gr
 import os 
 import json
-from transformers import pipeline
 
 # Set your API key
 palm.configure(api_key=os.environ['PALM_KEY'])
@@ -10,13 +9,13 @@ palm.configure(api_key=os.environ['PALM_KEY'])
 # Select the PaLM 2 model
 # model = 'models/text-bison-001'
 
-# candlab= ["recharge coins or get subscription", "show list of my friends"]
 
 
 
 def responsenew(data):
     print(data)
-    response = palm.chat(messages=f"""From the text given as data below by the user, find out what intention or category does the data fall under out of given 5 intents i.e:\n
+    response = palm.chat(message=data)
+    intent = palm.chat(messages=f"""From the text given as data below by the user, find out what intention or category does the data fall under out of given 5 intents i.e:\n
     1. purchasing coins\n
     2. viewing friends list\n
     3. viewing groups been joined by the user\n
@@ -32,30 +31,48 @@ def responsenew(data):
     5. sharedrem\n\n
     Your answer must be of one word only out of these above given 5 words.""")
 
-    respo = {
-            "message": response.last,
-            "action": "nothing",
-            "function": "nothing"
-        }
-    # user_intent = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
-    # if "payment" in response.last:
-    #     respo = {
-    #         "message": "Click the button below to view Premium Services and Coin Recharge options: ",
-    #         "action": "payment",
-    #         "function": "nothing"
-    #     }
-    # elif "friends" in response.last:
-    #     respo = {
-    #         "message": "Slide left or right profiles or tap on 'My Friends' to view list of your friends: ",
-    #         "action": "show_friends",
-    #         "function": "nothing"
-    #     }
-    # else:
-    #     respo = {
+    # respo = {
     #         "message": response.last,
     #         "action": "nothing",
     #         "function": "nothing"
     #     }
+    
+    if "recoin" in intent.last:
+        respo = {
+            "message": "Click the button below to view Premium Services and Coin Recharge options: ",
+            "action": "payment",
+            "function": "nothing"
+        }
+    elif "view_friends" in intent.last:
+        respo = {
+            "message": "Here's the list of your friends: ",
+            "action": "show_friends",
+            "function": "nothing"
+        }
+    elif "view_groups" in intent.last:
+        respo = {
+            "message": "You are member of following groups: ",
+            "action": "show_mygroups",
+            "function": "nothing"
+        }
+    elif "view_pages" in intent.last:
+        respo = {
+            "message": "You are part of following communities🫶: ",
+            "action": "show_mycommunities",
+            "function": "nothing"
+        }
+    elif "sharedrem" in intent.last:
+        respo = {
+            "message": "Here's the list of your shared reminders: ",
+            "action": "shared_reminders",
+            "function": "nothing"
+        }
+    else:
+        respo = {
+            "message": response.last,
+            "action": "nothing",
+            "function": "nothing"
+        }
     return json.dumps(respo)
 
 gradio_interface = gr.Interface(