ChatGPT-Plugins-UI-with-Langchain / gpt_function_definitions.py
ysharma's picture
ysharma HF staff
added bored_api function def
ae72a48
raw
history blame contribute delete
No virus
4.49 kB
from newsapi import NewsApiClient
from gradio_client import Client
import requests
import random
# example input: prompt = "Beautiful Sky with "Gradio is love" written over it"
# defining a function to generate music using Gradio demo of TextDiffusers hosted on Spaces
def generate_image(prompt: str) -> str:
"""
generate an image based on the prompt provided
"""
client = Client("https://ysharma-textdiffuser.hf.space/", hf_token="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")
result = client.predict(
prompt, # str in 'Input your prompt here. Please enclose keywords with 'single quotes', you may refer to the examples below. The current version only supports input in English characters.' Textbox component
20, # int | float (numeric value between 1 and 50) in 'Sampling step' Slider component
7.5, # int | float (numeric value between 1 and 9) in 'Scale of classifier-free guidance' Slider component
1, # int | float (numeric value between 1 and 4) in 'Batch size' Slider component
"Stable Diffusion v2.1", # str in 'Pre-trained Model' Radio component
fn_index=1)
return result[0]
# example input: input_text = "A cheerful country song with acoustic guitars"
# defining a function to generate music using Gradio demo of MusicGen hosted on Spaces
#input melody example = "/content/bolero_ravel.mp3"
def generate_music(input_text, input_melody ):
"""
generate music based on an input text
"""
client = Client("https://ysharma-musicgendupe.hf.space/", hf_token="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")
result = client.predict(
"melody", # str in 'Model' Radio component
input_text, # str in 'Input Text' Textbox component
input_melody, # str (filepath or URL to file) in 'Melody Condition (optional)' Audio component
5, # int | float (numeric value between 1 and 120) in 'Duration' Slider component
250, # int | float in 'Top-k' Number component
0, # int | float in 'Top-p' Number component
1, # int | float in 'Temperature' Number component
3, # int | float in 'Classifier Free Guidance' Number component
fn_index=1)
return result
# example input: input_image = "cat.jpg"
# defining a function to generate caption using a image caption Gradio demo hosted on Spaces
def generate_caption(input_image ):
"""
generate caption for the input image
"""
client = Client("https://nielsr-comparing-captioning-models.hf.space/")
temp = input_image.split('/')
if len(temp) == 1:
input_image = temp[0]
else:
input_image = temp[-1]
result = client.predict(
input_image,
api_name="/predict")
result = "The image can have any one of the following captions, all captions are correct: " + ", or ".join([f"'{caption.replace('.','')}'" for caption in result])
return result
# defining a function to get the most relevant world news for a given query
# example query: Joe Biden presidency
def get_news(search_query):
"""
get top three news items for your search query
"""
newsapi = NewsApiClient(api_key='8c50d2b3d80348a2a78be021b2c49cd1')
docs = newsapi.get_everything(q=search_query,
language='en',
sort_by = 'relevancy',
page_size=3,
page=1
)['articles']
res = [news['description'] for news in docs]
res = [item.replace('<li>','').replace('</li>','').replace('<ol>','') for item in res]
res = "\n".join([f"{i}.{ res[i-1]}" for i in range(1,len(res)+1)])
return "Following list has the top three news items for the given search query : \n" + res
# example inputs: prompt = "I am bored what can I do?"
# defining a function to get an activity basee on activity type
# Activity type - education, recreational, social, DIY, charity, cooking, relaxation, music, and busywork.
def bored_api(activity_type) -> str:
"""
Get a random activity to do based on the activity type.
"""
activity_type_list = ["education", "recreational", "social", "diy", "charity", "cooking", "relaxation", "music", "busywork"]
activity_type = activity_type.lower()
if activity_type not in activity_type_list:
activity_type = random.choice(activity_type_list)
api_url = "https://www.boredapi.com/api/activity/?type=" + activity_type
response = requests.get(
api_url
)
return response.json()['activity']