|
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) |
|
|
|
st.set_page_config(page_title="Chatbot", layout="centered") |
|
st.title("π¬ Chat with LLM") |
|
|
|
|
|
if "history" not in st.session_state: |
|
st.session_state.history = [] |
|
|
|
|
|
user_input = st.text_input("Ask me anything:", key="input") |
|
|
|
|
|
if user_input: |
|
|
|
st.session_state.history.append(("You", user_input)) |
|
|
|
|
|
response = llm.invoke(user_input) |
|
|
|
|
|
st.session_state.history.append(("Bot", response.content)) |
|
|
|
|
|
for sender, msg in st.session_state.history: |
|
st.markdown(f"**{sender}:** {msg}") |
|
|