Update app.py

app.py

@@ -3,48 +3,36 @@ from dotenv import load_dotenv
 import os
 from langchain_openai import ChatOpenAI
 
-# Load environment variables
+# Load API keys
 load_dotenv("apiroute.env")
 api_key = os.getenv("OPENAI_API_KEY")
 api_base = os.getenv("OPENAI_API_BASE")
 
-# Sanity check
-if not api_key or not api_base:
-    st.error("❌ API key or base URL missing in apiroute.env.")
-    st.stop()
-
 
 
 # Init LLM
 llm = ChatOpenAI(model_name="google/gemma-3n-e2b-it:free", temperature=0.7)
 
-# Page settings
+# Streamlit UI
 st.set_page_config(page_title="Chatbot", layout="centered")
-st.title("💬 Chat with me")
+st.title("💬 Chat with Gremmy")
 
-# Chat history
+# Session for chat history
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# Display chat history
+# Display chat history at top
 for sender, msg in st.session_state.history:
     st.markdown(f"**{sender}:** {msg}")
 
-# Input at bottom
-user_input = st.text_input("AsK me Anything", key="input")
+# Input with "Send" button
+with st.form(key="chat_form", clear_on_submit=True):
+    user_input = st.text_input("talk to me", key="user_message")
+    submitted = st.form_submit_button("Send")
 
-# Handle input
-if user_input and st.session_state.get("input_submitted") is not True:
+if submitted and user_input:
     st.session_state.history.append(("You", user_input))
     response = llm.invoke(user_input)
-    st.session_state.history.append(("Bot", response.content))
-
-    # ✅ Mark input as handled and clear box
-    st.session_state.input_submitted = True
-    st.session_state.input = ""  # Clear input box
+    st.session_state.history.append(("Gremmy", response.content))
     st.rerun()
 
-# ✅ Reset flag after rerun
-if st.session_state.get("input_submitted"):
-    st.session_state.input_submitted = False
-