", unsafe_allow_html=True)
st.header(t['ask_ai'])
# Display chat history
with st.container():
st.markdown("
", unsafe_allow_html=True)
if not st.session_state.chat_history:
st.markdown("Say hello! I'm your Timescope Assistant, ready to help you explore news and history.")
for msg in st.session_state.chat_history:
role_label = "🧑💻 You" if msg["role"] == "user" else "🤖 Assistant"
st.markdown(f"**{role_label}:** {translate_text(msg['content'], lang_code)}") # Translate history too
st.markdown("
", unsafe_allow_html=True)
# Input form for the AI assistant
with st.form("chat_form", clear_on_submit=True):
user_input = st.text_input(t['ask_placeholder'],
key="assistant_input",
label_visibility="collapsed",
placeholder=t['ask_placeholder']) # Use translated placeholder
send_button = st.form_submit_button("Send")
if send_button and user_input:
# Add user message to history
st.session_state.chat_history.append({"role": "user", "content": user_input})
with st.spinner("Thinking..."):
try:
# Check if API key is truly available before making the call
if not openrouter_api_key or openrouter_api_key == "sk-dummy-key-for-no-ai":
st.error(f"{t['ai_error']} OpenRouter API Key is not set. Please add it to your `.env` file.")
else:
# Build the messages list first
messages = [
{"role": "system", "content": translate_text('You are a helpful assistant about world news and history.', lang_code)}
] + [
{"role": m["role"], "content": m["content"]} for m in st.session_state.chat_history
]
resp = client.chat.completions.create(
model=DEFAULT_AI_MODEL, # Use the OpenRouter model
messages=messages,
max_tokens=1000, # Max tokens to control response length and cost
)
assistant_response = resp.choices[0].message.content.strip()
st.session_state.chat_history.append({"role": "assistant", "content": assistant_response}) # Store original response
# No need to display directly here, it will be rerendered by st.markdown in history display
st.rerun() # Rerun to update chat history display
except Exception as e:
if isinstance(e, RateLimitError):
st.error(f"{t['ai_error']} You've hit a rate limit. Please wait a moment and try again.")
st.error(f"{t['ai_error']} You've hit a rate limit. Please wait a moment and try again.")
elif isinstance(e, AuthenticationError):
st.error(f"{t['ai_error']} Authentication failed. Please check your OpenRouter API key in your `.env` file.")
elif isinstance(e, APIConnectionError):
st.error(f"{t['ai_error']} Could not connect to the AI service. Please check your internet connection or the OpenRouter API status.")
elif isinstance(e, APIStatusError):
# This handles the 402 error for credits as well as other API errors
st.error(f"{t['ai_error']} An API error occurred: {e.status_code} - {e.response.text if hasattr(e.response, 'text') else str(e)}")
else:
st.error(f"{t['ai_error']} An unexpected error occurred: {e}")
st.session_state.chat_history.append({"role": "assistant", "content": f"Error: {e}"}) # Add error to history
st.rerun() # Rerun to display error in history
st.markdown("