Update app.py
Browse files
app.py
CHANGED
|
@@ -8,35 +8,43 @@ load_dotenv("apiroute.env")
|
|
| 8 |
api_key = os.getenv("OPENAI_API_KEY")
|
| 9 |
api_base = os.getenv("OPENAI_API_BASE")
|
| 10 |
|
| 11 |
-
#
|
| 12 |
if not api_key or not api_base:
|
| 13 |
-
st.error("β
|
| 14 |
st.stop()
|
| 15 |
|
| 16 |
-
# Set for LangChain/OpenAI use
|
| 17 |
-
os.environ["OPENAI_API_KEY"] = api_key
|
| 18 |
-
os.environ["OPENAI_API_BASE"] = api_base
|
| 19 |
|
| 20 |
-
|
|
|
|
| 21 |
llm = ChatOpenAI(model_name="google/gemma-3n-e2b-it:free", temperature=0.7)
|
| 22 |
|
| 23 |
-
#
|
| 24 |
st.set_page_config(page_title="Chatbot", layout="centered")
|
| 25 |
-
st.title("π¬ Chat with
|
| 26 |
-
|
| 27 |
|
|
|
|
| 28 |
if "history" not in st.session_state:
|
| 29 |
st.session_state.history = []
|
| 30 |
|
| 31 |
-
|
| 32 |
for sender, msg in st.session_state.history:
|
| 33 |
st.markdown(f"**{sender}:** {msg}")
|
| 34 |
|
| 35 |
-
|
| 36 |
-
|
| 37 |
|
| 38 |
-
|
|
|
|
| 39 |
st.session_state.history.append(("You", user_input))
|
| 40 |
response = llm.invoke(user_input)
|
| 41 |
st.session_state.history.append(("Bot", response.content))
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
api_key = os.getenv("OPENAI_API_KEY")
|
| 9 |
api_base = os.getenv("OPENAI_API_BASE")
|
| 10 |
|
| 11 |
+
# Sanity check
|
| 12 |
if not api_key or not api_base:
|
| 13 |
+
st.error("β API key or base URL missing in apiroute.env.")
|
| 14 |
st.stop()
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
+
|
| 18 |
+
# Init LLM
|
| 19 |
llm = ChatOpenAI(model_name="google/gemma-3n-e2b-it:free", temperature=0.7)
|
| 20 |
|
| 21 |
+
# Page settings
|
| 22 |
st.set_page_config(page_title="Chatbot", layout="centered")
|
| 23 |
+
st.title("π¬ Chat with me")
|
|
|
|
| 24 |
|
| 25 |
+
# Chat history
|
| 26 |
if "history" not in st.session_state:
|
| 27 |
st.session_state.history = []
|
| 28 |
|
| 29 |
+
# Display chat history
|
| 30 |
for sender, msg in st.session_state.history:
|
| 31 |
st.markdown(f"**{sender}:** {msg}")
|
| 32 |
|
| 33 |
+
# Input at bottom
|
| 34 |
+
user_input = st.text_input("AsK me Anything", key="input")
|
| 35 |
|
| 36 |
+
# Handle input
|
| 37 |
+
if user_input and st.session_state.get("input_submitted") is not True:
|
| 38 |
st.session_state.history.append(("You", user_input))
|
| 39 |
response = llm.invoke(user_input)
|
| 40 |
st.session_state.history.append(("Bot", response.content))
|
| 41 |
+
|
| 42 |
+
# β
Mark input as handled and clear box
|
| 43 |
+
st.session_state.input_submitted = True
|
| 44 |
+
st.session_state.input = "" # Clear input box
|
| 45 |
+
st.rerun()
|
| 46 |
+
|
| 47 |
+
# β
Reset flag after rerun
|
| 48 |
+
if st.session_state.get("input_submitted"):
|
| 49 |
+
st.session_state.input_submitted = False
|
| 50 |
+
|