Spaces:
Runtime error
Runtime error
File size: 6,654 Bytes
374c0fd df325aa 374c0fd df325aa 374c0fd |
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 131 132 133 |
import streamlit as st
from user_roles import user_role_list
from promptTemplate import task_generator
from promptTemplate import conversation_generator
import openai
import json
# page title
st.set_page_config(page_title="LLM Roleplay")
# sidebar content
with st.sidebar:
st.markdown("My side bar")
# main function
def main():
st.title("LLM Roleplay")
if "submit" not in st.session_state:
st.session_state.submit = False
if "submit2" not in st.session_state:
st.session_state.submit2 = False
if "initial" not in st.session_state:
st.session_state.initial = False
if "initial2" not in st.session_state:
st.session_state.initial2 = False
if "extracted" not in st.session_state:
st.session_state.extracted = False
with st.form(key="form"):
role1, role2 = st.columns([0.5, 0.5])
with role1:
# role selection dropdown menu
selected_roles_user = st.multiselect("Select User Roles", user_role_list)
user = ', '.join(selected_roles_user)
# user = st.text_input("User Role (EDIT ME)", value=', '.join(selected_roles_user))
with role2:
# role selection dropdown menu
selected_roles_assistant = st.multiselect("Select Assistant Roles", user_role_list)
assistant = ', '.join(selected_roles_assistant)
# assistant = st.text_input("Assistant Role (EDIT ME)", value=', '.join(selected_roles_assistant))
# input for chat_limit
chat_limit = st.number_input(
"Messages to generate:", min_value=1, max_value=20, value=5
)
# input text for openAiKey
openAiKey = st.text_input(label="Input the openai key", type="password")
# form submit button and setting up the session_state
if st.form_submit_button(label="Submit"):
st.session_state.submit = True
st.session_state.initial = True
# print(user)
# print(assistant)
# print(prompt(user, assistant))
if st.session_state.submit:
print("Inside submit")
with st.spinner("In progress..."):
try:
if st.session_state.initial:
st.session_state["task"] = task_generator(user, assistant, openAiKey)
print("task_generator")
# print("\n\n\nAfter the task\n\n\n")
st.session_state.initial = False
with st.form(key="form2"):
# input text for task
st.session_state["task"] = st.text_input("The topic for the roleplay (Edit if required)", value=st.session_state["task"])
if st.form_submit_button(label="Generate"):
st.session_state.submit2 = True
# print(st.session_state["task"])
st.session_state.initial2 = True
# print("\n\nsubmit2 statues:")
# print(st.session_state.submit2)
# print("\n\n")
except Exception as err:
st.error(err)
if st.session_state.submit2:
print("Inside submit2")
st.session_state.extracted = False
with st.expander(label="role play"):
with st.spinner("Generating..."):
try:
if st.session_state.initial2:
roleplay, user_convo, assistant_convo = conversation_generator(user, assistant, st.session_state["task"],chat_limit)
# roleplay = {'user': 'Actor', 'assistant': 'Chef', 'task': "Planning and preparing healthy meals that meet the actor's strict dietary requirements.", 'conversations': [{'from': 'user', 'value': 'Create a list of movie genres that will inspire each course of the menu.'}, {'from': 'assistant', 'value': 'To create a list of movie genres that will inspire each course of the menu, we can consider a variety of genres that offer distinct flavors and themes. Here is a suggested list:\n\n1. Sci-Fi: Molecular Gastronomy Appetizer\n2. Romance: Elegant Seafood Entrée\n3. Action: Spicy Asian Fusion Main Course\n4. Comedy: Playful Dessert\n5. Western: Smoked BBQ Main Course'}, {'from': 'user', 'value': 'Choose a specific sci-fi movie that will inspire the molecular gastronomy appetizer.'}, {'from': 'assistant', 'value': 'To choose a specific sci-fi movie that will inspire the molecular gastronomy appetizer, we can look for a movie that showcases futuristic technology, innovative concepts, and unique visuals. One movie that fits this description is "Blade Runner." The dystopian setting and advanced technology in the movie can serve as inspiration for creating a visually stunning and futuristic molecular gastronomy appetizer.'}], 'specified_task': 'Chef will help Actor create a delectable five-course menu inspired by movie genres. From a sci-fi-inspired molecular gastronomy appetizer to a western-themed smoked BBQ main course, Chef will bring the flavors and ambiance of each genre to life, ensuring a memorable dining experience for Actor and their guests.', 'length': 4}
st.session_state["roleplay"] = roleplay
st.session_state.initial2 = False
print("conversation_generator")
# print("roleplay")
# print(roleplay)
# print("user_convo")
# print(user_convo)
# print("assistant_convo")
# print(assistant_convo)
st.session_state.extracted = True
except Exception as err:
st.error(err)
if st.session_state.extracted:
json_data = st.session_state["roleplay"]
json_string = json.dumps(json_data, ensure_ascii=False)
if st.download_button(label="Save as jsonl", data=json_string , mime="application/json"):
st.success("Successfully saved")
if st.button("Clear"):
st.session_state.submit = False
st.session_state.submit2 = False
st.session_state.initial = False
st.session_state.initial2 = False
st.session_state.extracted = False
if "task" in st.session_state:
del st.session_state["task"]
if "roleplay" in st.session_state:
del st.session_state["roleplay"]
st.experimental_rerun()
if __name__ == "__main__":
main() |