Spaces:
Runtime error
Runtime error
| 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() |