"""
history_html += bot_html_start
yield history_html
# 4. Stream the response text
current_text = ""
for char in response_text:
current_text += char
# Update the last span in history_html
# We reconstruct the HTML for the streaming part
streaming_content = f"""
{agent['avatar']} {agent['name']}:
{current_text}|
"""
# To do this properly in Gradio HTML streaming, we usually replace the whole content or append.
# Here, we will append the text chunk by chunk.
# A simpler approach for HTML streaming: Just yield the accumulated text.
# Replace the last open streaming-text tag with the new content
# This is a bit tricky with string manipulation, so we'll just append for visual effect
# or yield the full history. Let's yield the full history.
# Reconstruct full history for this step
# (In a production app with huge history, we'd use JS to append, but for Python generator):
# We will just yield the full string for simplicity and robustness.
# Actually, let's just append the char to a buffer and yield the bot message HTML
pass
# Better Streaming Logic for HTML Component:
# We yield the message piece by piece.
for i in range(len(response_text)):
chunk = response_text[:i+1]
# Construct the specific bot message HTML
bot_chunk_html = f"""
{agent['avatar']} {agent['name']}:
{chunk}|
"""
# We need to append this to the main history.
# Since we can't easily "append" to the previous yield without state,
# we will assume the generator logic is:
# Yield User -> Yield Bot Start -> Yield Bot Chunks -> Yield Bot End
# For this implementation, let's keep it simple: Standard streaming text update
# But we are using HTML component.
# Let's revert to: Generate full response, then stream the HTML string appending chars.
# SIMPLIFIED STREAMING STRATEGY:
# 1. User msg
# 2. Bot msg wrapper
# 3. Loop chars -> yield updated HTML string
# Since we can't easily modify the *previous* yield's content in Python (only replace output),
# We will build the string incrementally.
# Let's restart the generator logic for clarity:
pass
# RE-IMPLEMENTING GENERATOR FOR CLARITY:
# We will take the current history, add user, then add bot progressively.
# Note: The actual implementation below handles the string building correctly.
pass
# Final robust generator
def quantum_chat_stream(message: str, history_html: str, agent_key: str) -> Generator[str, None, None]:
if not message:
yield history_html
return
# 1. User Message
user_msg = f"""
YOU: {message}
{datetime.now().strftime('%H:%M')}
"""
history_html += user_msg
yield history_html
# 2. Get Response
response = quantum_bot.process_message(message, agent_key)
agent = quantum_bot.personalities[agent_key]
# 3. Bot Message Start
bot_header = f"""
{agent['avatar']} {agent['name']}:
"""
history_html += bot_header
yield history_html
# 4. Stream Content
accumulated_text = ""
for char in response:
accumulated_text += char
# Append the character to the history.
# We must close the divs every time or use a specific span.
# Let's use a span for the text content.
content_span = f'{accumulated_text}|
'
# Remove the last closing div if we added it previously?
# No, simpler: Just yield the history + accumulated text.
# But we need to close the div properly.
# Current history has the opening div + header.
# We need to append the span + closing div.
yield history_html + content_span
# 5. Final clean up (remove cursor)
final_bot_msg = f'{accumulated_text}