LLm_file / app.py
Enoch1359's picture
Update app.py
64a3019 verified
raw
history blame
1.02 kB
import streamlit as st
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
import os
load_dotenv("apiroute.env")
api_key=os.getenv("OPENAI_API_KEY")
api_base=os.getenv("OPENAI_API_BASE")
os.environ["OPENAI_API_KEY"]=api_key
os.environ['OPENAI_API_BASE']=api_base
llm=ChatOpenAI(model_name="google/gemma-3n-e2b-it:free",temperature=0.7)
# Streamlit UI
st.set_page_config(page_title="Chatbot", layout="centered")
st.title("πŸ’¬ Chat with LLM")
# Chat history session state
if "history" not in st.session_state:
st.session_state.history = []
# Prompt input
user_input = st.text_input("Ask me anything:", key="input")
# Handle input
if user_input:
# Append user message
st.session_state.history.append(("You", user_input))
# Get response from LLM
response = llm.invoke(user_input)
# Append bot message
st.session_state.history.append(("Bot", response.content))
# Display conversation
for sender, msg in st.session_state.history:
st.markdown(f"**{sender}:** {msg}")