Synced repo using 'sync_with_huggingface' Github Action

Roleplay/app.py

@@ -44,8 +44,10 @@ def main():
             assistant = ', '.join(selected_roles_assistant)
             # assistant = st.text_input("Assistant Role (EDIT ME)", value=', '.join(selected_roles_assistant))
 
-        # input text for task
-        # task = st.text_input("Topic")
+        # 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")
@@ -93,7 +95,7 @@ def main():
                 try:
                     if st.session_state.initial2:
                         
-                        roleplay, user_convo, assistant_convo = conversation_generator(user, assistant, st.session_state["task"])
+                        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

Roleplay/promptTemplate.py

@@ -36,6 +36,7 @@ def conversation_generator(user,assistant, task, chat_limit=2):
     print(task)
     print(user)
     print(assistant)
+    print(chat_limit)
     role_play_session = RolePlaying(assistant, user, task_prompt)
     # print(Fore.CYAN + f"Specified task prompt:\n{role_play_session.task_prompt}\n")
     # print(f"Specified task prompt:\n{role_play_session.task_prompt}\n")