Update app.py

app.py

@@ -1,34 +1,42 @@
 import streamlit as st
 from dotenv import load_dotenv
-from langchain_openai import ChatOpenAI
 import os
+from langchain_openai import ChatOpenAI
+
+# Load environment variables
 load_dotenv("apiroute.env")
-api_key=os.getenv("OPENAI_API_KEY")
-api_base=os.getenv("OPENAI_API_BASE")
+api_key = os.getenv("OPENAI_API_KEY")
+api_base = os.getenv("OPENAI_API_BASE")
+
+# Ensure they exist
+if not api_key or not api_base:
+    st.error("❌ Missing API credentials in apiroute.env.")
+    st.stop()
+
+# Set for LangChain/OpenAI use
+os.environ["OPENAI_API_KEY"] = api_key
+os.environ["OPENAI_API_BASE"] = api_base
+
+# Initialize LLM
+llm = ChatOpenAI(model_name="google/gemma-3n-e2b-it:free", temperature=0.7)
 
-llm=ChatOpenAI(model_name="google/gemma-3n-e2b-it:free",temperature=0.7)
-# Streamlit UI
+# Streamlit setup
 st.set_page_config(page_title="Chatbot", layout="centered")
-st.title("💬 Chat with me")
+st.title("💬 Chat with LLM")
+
 
-# Chat history session state
 if "history" not in st.session_state:
     st.session_state.history = []
 
 
-#Then place the prompt box at the bottom
-user_input = st.text_input("Ask me anything:", key="input")
+for sender, msg in st.session_state.history:
+    st.markdown(f"**{sender}:** {msg}")
+
+user_input = st.text_input("Type your message and press Enter", key="input")
+
 
-# Handle input
 if user_input:
-    # Add user message to history
     st.session_state.history.append(("You", user_input))
-
-    # Get response from the model
     response = llm.invoke(user_input)
-
-    # Add bot response to history
     st.session_state.history.append(("Bot", response.content))
-
-    # Clear the input box after submission
-    st.experimental_rerun()  
+    st.experimental_rerun()  # force UI refresh