Spaces:
Runtime error
Runtime error
File size: 4,674 Bytes
60f5bcc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
import streamlit as st
from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate
from langchain import FewShotPromptTemplate
from langchain.prompts.example_selector import LengthBasedExampleSelector
from dotenv import load_dotenv
load_dotenv()
def getLLMResponse(query, age_option, tasktype_option):
llm = OpenAI(temperature=0.9, model="text-davinci-003")
examples = [
{
"query": "What is a mobile?",
"answer": "A mobile is a magical device that fits in your pocket, like a mini-enchanted playground. It has games, videos, and talking pictures, but be careful, it can turn grown-ups into screen-time monsters too!",
},
{
"query": "What are your dreams?",
"answer": "My dreams are like colorful adventures, where I become a superhero and save the day! I dream of giggles, ice cream parties, and having a pet dragon named Sparkles..",
},
{
"query": " What are your ambitions?",
"answer": "I want to be a super funny comedian, spreading laughter everywhere I go! I also want to be a master cookie baker and a professional blanket fort builder. Being mischievous and sweet is just my bonus superpower!",
},
{
"query": "What happens when you get sick?",
"answer": "When I get sick, it's like a sneaky monster visits. I feel tired, sniffly, and need lots of cuddles. But don't worry, with medicine, rest, and love, I bounce back to being a mischievous sweetheart!",
},
{
"query": "WHow much do you love your dad?",
"answer": "Oh, I love my dad to the moon and back, with sprinkles and unicorns on top! He's my superhero, my partner in silly adventures, and the one who gives the best tickles and hugs!",
},
{
"query": "Tell me about your friend?",
"answer": "My friend is like a sunshine rainbow! We laugh, play, and have magical parties together. They always listen, share their toys, and make me feel special. Friendship is the best adventure!",
},
{
"query": "What math means to you?",
"answer": "Math is like a puzzle game, full of numbers and shapes. It helps me count my toys, build towers, and share treats equally. It's fun and makes my brain sparkle!",
},
{
"query": "What is your fear?",
"answer": "Sometimes I'm scared of thunderstorms and monsters under my bed. But with my teddy bear by my side and lots of cuddles, I feel safe and brave again!",
},
]
example_template = """
Question: {query}
Response: {answer}
"""
example_prompt = PromptTemplate(
input_variables=["query", "answer"], template=example_template
)
prefix = """You are a {template_ageoption}, and {template_tasktype_option}:
Here are some examples:
"""
suffix = """
Question: {template_userInput}
Response: """
example_selector = LengthBasedExampleSelector(
examples=examples, example_prompt=example_prompt, max_length=200
)
new_prompt_template = FewShotPromptTemplate(
example_selector=example_selector, # use example_selector instead of examples
example_prompt=example_prompt,
prefix=prefix,
suffix=suffix,
input_variables=[
"template_userInput",
"template_ageoption",
"template_tasktype_option",
],
example_separator="\n",
)
print(
new_prompt_template.format(
template_userInput=query,
template_ageoption=age_option,
template_tasktype_option=tasktype_option,
)
)
response = llm(
new_prompt_template.format(
template_userInput=query,
template_ageoption=age_option,
template_tasktype_option=tasktype_option,
)
)
print(response)
return response
# UI Starts here
st.set_page_config(
page_title="Marketing Tool",
page_icon="✅",
layout="centered",
initial_sidebar_state="collapsed",
)
st.header("Hey, How can I help you?")
form_input = st.text_area("Enter text", height=275)
tasktype_option = st.selectbox(
"Please select the action to be performed?",
("Write a sales copy", "Create a tweet", "Write a product description"),
key=1,
)
age_option = st.selectbox(
"For which age group?", ("Kid", "Adult", "senior Citizen"), key=2
)
numberOfWords = st.slider("Words limit", 1, 200, 25)
submit = st.button("Generate")
if submit:
st.write(getLLMResponse(form_input, tasktype_option, age_option))
|