app.py

import google.generativeai as palm
import gradio as gr
import os 
import json

# Set your API key
palm.configure(api_key=os.environ['PALM_KEY'])

# Select the PaLM 2 model
# model = 'models/text-bison-001'




def responsenew(data):
    print(data)
    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
    4. viewing pages been joined by the user\n
    5. user is saying to view the reminders been shared to the user or by the user\n
    
    data = {data}\n\n
    After you are done find out the intent, answer in one word only the intent. Use the following word for your answer, as given below in sequence to the intent:\n
    1. recoin\n
    2. view_friends\n
    3. view_groups\n
    4. view_pages\n
    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"
    #     }
    
    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(
  fn = responsenew,
  inputs = "text",
  outputs = "text" 
)
gradio_interface.launch()