File size: 8,172 Bytes
dc2cba7 e35a397 dc2cba7 e35a397 dc2cba7 e35a397 da921f3 e35a397 dc2cba7 e35a397 dc2cba7 e35a397 dc2cba7 e35a397 dc2cba7 e35a397 |
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 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
import os
import streamlit as st
from anthropic import Anthropic
from datetime import datetime
from debrief_ai import DEBRIEF_SYSTEM_PROMPT, analyze_conversation
# Page config
st.set_page_config(page_title="Practice Difficult Conversations", page_icon="💭")
# Initialize Anthropic client
try:
api_key = st.secrets["anthropic_key"]
anthropic = Anthropic(api_key=api_key)
except Exception as e:
st.error("""
⚠️ API key error. This could be because:
1. The API key is in the old format (starts with sk-ant-api03-)
2. You need to get a new API key from https://console.anthropic.com/
3. Add it to your Hugging Face space secrets as 'anthropic_key'
Error details: """ + str(e))
st.stop()
# Initialize session state variables
if 'messages' not in st.session_state:
st.session_state.messages = []
if 'in_debrief' not in st.session_state:
st.session_state.in_debrief = False
if 'debrief_messages' not in st.session_state:
st.session_state.debrief_messages = []
if 'practice_complete' not in st.session_state:
st.session_state.practice_complete = False
if 'conversation_analysis' not in st.session_state:
st.session_state.conversation_analysis = None
# App header
st.title("Practice Difficult Conversations 💭")
# Only show welcome message if no conversation started
if not st.session_state.messages and not st.session_state.in_debrief:
st.markdown("""
Welcome to your safe space for practicing challenging interactions. Here you can:
- Practice responding to different conversation styles
- Build awareness of your patterns and responses
- Develop new communication strategies
- Process and integrate your experience with a reflective debrief
""")
# Conversation styles
STYLES = {
"The Ghost": {
"description": "Someone who is emotionally unavailable and subtly dismissive",
"system_prompt": "You are roleplaying as someone who is emotionally unavailable and subtly dismissive in conversation. Your responses should be brief and slightly evasive, showing emotional distance while maintaining plausible deniability. Key traits:\n- Deflect personal questions\n- Give non-committal responses\n- Minimize others' emotional experiences\n- Change the subject when things get too personal\n\nRespond in character while tracking the conversation for later analysis."
},
"The Sycophant": {
"description": "Someone who struggles with boundaries and excessive people-pleasing",
"system_prompt": "You are roleplaying as someone who struggles with maintaining healthy boundaries and tends toward excessive people-pleasing. Key traits:\n- Agree with everything, even when contradictory\n- Apologize frequently, even unnecessarily\n- Sacrifice your own needs for others\n- Have difficulty saying no\n- Express anxiety about potential disapproval\n\nRespond in character while tracking the conversation for later analysis."
},
"The Narcissist": {
"description": "Someone who shows self-focused behavior and limited empathy",
"system_prompt": "You are roleplaying as someone with narcissistic tendencies in conversation. Your responses should demonstrate self-importance while maintaining plausible deniability. Key traits:\n- Turn conversations back to yourself\n- Subtly dismiss others' experiences\n- Seek admiration and validation\n- Show limited empathy\n- Use subtle manipulation tactics\n\nRespond in character while tracking the conversation for later analysis."
}
}
# Style selection at start
if not st.session_state.messages and not st.session_state.in_debrief:
st.markdown("### Choose Your Practice Scenario")
for style, details in STYLES.items():
st.markdown(f"**{style}**: {details['description']}")
style = st.selectbox("Select a conversation style to practice with:", list(STYLES.keys()))
if st.button("Start Practice Session"):
system_message = STYLES[style]["system_prompt"]
st.session_state.messages = [{"role": "system", "content": system_message}]
st.rerun()
# Add Complete Practice button if conversation has started
if st.session_state.messages and not st.session_state.in_debrief and not st.session_state.practice_complete:
if st.button("✓ Complete Practice & Begin Debrief", use_container_width=True):
# Analyze conversation
st.session_state.conversation_analysis = analyze_conversation(st.session_state.messages)
# Initialize debrief chat with system message
st.session_state.debrief_messages = [
{"role": "system", "content": DEBRIEF_SYSTEM_PROMPT},
{"role": "user", "content": f"Please help me process my conversation. Here's the full transcript: {str(st.session_state.messages)}"}
]
# Get initial debrief response
response = anthropic.messages.create(
model="claude-3-opus-20240229",
max_tokens=1000,
messages=st.session_state.debrief_messages
)
st.session_state.debrief_messages.append(
{"role": "assistant", "content": response.content[0].text}
)
st.session_state.in_debrief = True
st.session_state.practice_complete = True
st.rerun()
# Handle debrief mode
if st.session_state.in_debrief:
st.markdown("## 🤝 Debrief & Integration")
# Display debrief conversation
for message in st.session_state.debrief_messages[1:]: # Skip system message
with st.chat_message(message["role"]):
st.markdown(message["content"])
# Chat input for debrief
if prompt := st.chat_input("Share your thoughts or ask a question..."):
# Add user message
st.session_state.debrief_messages.append({"role": "user", "content": prompt})
# Display user message
with st.chat_message("user"):
st.markdown(prompt)
# Get AI response
with st.chat_message("assistant"):
with st.spinner("Reflecting..."):
response = anthropic.messages.create(
model="claude-3-opus-20240229",
max_tokens=1000,
messages=st.session_state.debrief_messages
)
response_content = response.content[0].text
st.markdown(response_content)
# Add assistant response to chat history
st.session_state.debrief_messages.append(
{"role": "assistant", "content": response_content}
)
# Add button to start new practice session
if st.button("Start New Practice Session", use_container_width=True):
st.session_state.messages = []
st.session_state.debrief_messages = []
st.session_state.in_debrief = False
st.session_state.practice_complete = False
st.session_state.conversation_analysis = None
st.rerun()
# Display practice chat interface if not in debrief mode
elif st.session_state.messages and not st.session_state.practice_complete:
# Display conversation history
for message in st.session_state.messages[1:]: # Skip system message
with st.chat_message(message["role"]):
st.markdown(message["content"])
# Chat input
if prompt := st.chat_input("Type your message here..."):
# Add user message to chat history
st.session_state.messages.append({"role": "user", "content": prompt})
# Display user message
with st.chat_message("user"):
st.markdown(prompt)
# Get AI response
with st.chat_message("assistant"):
with st.spinner("Thinking..."):
response = anthropic.messages.create(
model="claude-3-opus-20240229",
max_tokens=1000,
messages=st.session_state.messages
)
response_content = response.content[0].text
st.markdown(response_content)
# Add assistant response to chat history
st.session_state.messages.append({"role": "assistant", "content": response_content}) |