Editing Openai/694ab010-3e7c-8000-b0e8-b98c22b7a063 (section)

=== Use these exact files in ~/ml-chatbot/. ===

==== <syntaxhighlight lang="python">import os ====
import streamlit as st
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage, SystemMessage, AIMessage

st.set_page_config(page_title="ML Chatbot", layout="centered")
st.title("Simple ML Chatbot (LangChain + Streamlit)")

api_key = os.getenv("OPENAI_API_KEY", "")
if not api_key:
    st.warning("OPENAI_API_KEY is not set. Set it as an environment variable for the container.")

if "history" not in st.session_state:
    st.session_state.history = [SystemMessage(content="You are a helpful assistant.")]

prompt = st.chat_input("Ask something...")

if prompt:
    st.session_state.history.append(HumanMessage(content=prompt))

    if api_key:
        llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.2, api_key=api_key)
        resp = llm.invoke(st.session_state.history)
        st.session_state.history.append(AIMessage(content=resp.content))
    else:
        st.session_state.history.append(AIMessage(content="Missing OPENAI_API_KEY. Set it and restart."))

for msg in st.session_state.history:
    if isinstance(msg, SystemMessage):
        continue
    role = "user" if isinstance(msg, HumanMessage) else "assistant"
    with st.chat_message(role):
        st.write(msg.content)

</syntaxhighlight>

==== <syntaxhighlight lang="txt">streamlit==1.39.0 ====
langchain==0.2.16
langchain-openai==0.1.22

</syntaxhighlight>

==== <syntaxhighlight lang="dockerfile">FROM python:3.11-slim ====

WORKDIR /app

RUN apt-get update && apt-get install -y --no-install-recommends \
    curl \
  && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

COPY app.py .

EXPOSE 8501

CMD ["streamlit", "run", "app.py", "--server.address=0.0.0.0", "--server.port=8501"]

</syntaxhighlight>