Spaces:
Sleeping
Sleeping
class ConversationProcessor: | |
def __init__(self, session_state): | |
self.session_state = session_state | |
def save_conversation_to_text(self, output_file): | |
with open(output_file, 'w', encoding='utf-8') as file: | |
file.write("Assistant: " + "\n") | |
for message in self.session_state['messages']: | |
role = message['role'] | |
content = message['content'] | |
if role == 'user': | |
file.write("User: " + content + "\n") | |
elif role == 'assistant': | |
file.write("Assistant: " + content + "\n") |