Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,25 +1,26 @@
+import os
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
-import os
 
-# Set OpenRouter API Key and Base URL
+# Get API key from environment variables
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 
-if not api_key:
-    st.error("API key not found. Please set OPENAI_API_KEY in Space secrets.")
+# Check if API key is available
+if not OPENAI_API_KEY:
+    st.error("🔑 API key not found. Please set OPENAI_API_KEY in Space secrets.")
     st.stop()
 
 # Set up the model
 llm = ChatOpenAI(
     temperature=0.7,
     model_name="deepseek/deepseek-chat-v3-0324:free",
-    openai_api_key=OPENAI_API_KEY,
+    openai_api_key=OPENAI_API_KEY,  # Use the variable here
     openai_api_base="https://openrouter.ai/api/v1"
 )
 
-# Set up memory
+# Rest of your chatbot code...
 if "memory" not in st.session_state:
     st.session_state.memory = ConversationBufferMemory()
 
@@ -31,7 +32,6 @@ conversation = ConversationChain(
 
 # Streamlit UI
 st.set_page_config(page_title="LLM Chatbot", page_icon="🤖")
-
 st.title("Langchain Chatbot by Muhammad Izhan")
 
 user_input = st.text_input("You:", key="input")
@@ -42,8 +42,7 @@ if user_input:
     st.session_state.memory.chat_memory.add_ai_message(response)
     st.write(f"**Bot:** {response}")
 
-# Show chat history
 if st.checkbox("Show Chat History"):
     for message in st.session_state.memory.chat_memory.messages:
         role = "You" if message.type == "human" else "Bot"
-        st.markdown(f"**{role}:** {message.content}")
+        st.markdown(f"**{role}:** {message.content}")