init

.gitignore

@@ -0,0 +1,3 @@
+.streamlit/
+.vscode/
+*.pyc

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+ENV PYTHONUNBUFFERED=1
+
+ENV VIRTUAL_ENV=/app/venv
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+RUN python -m venv ${VIRTUAL_ENV}
+
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+COPY . .
+
+EXPOSE 8501
+
+CMD ["streamlit", "run", "--server.port", "8501", "main.py"]

app.py

@@ -0,0 +1,89 @@
+import openai 
+import streamlit as st
+import uuid
+import time
+
+openai.api_key = st.secrets["OPENAI_API_KEY"]
+assistant_id = st.secrets["OPENAI_ASSISTANT_ID"]
+client = openai
+
+if "start_chat" not in st.session_state:
+    st.session_state.start_chat = False
+if "session_id" not in st.session_state:
+    st.session_state.session_id = str(uuid.uuid4())
+if "thread_id" not in st.session_state:
+    st.session_state.thread_id = None
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+st.set_page_config(page_title="Assistant API Chat", page_icon=":speech_balloon:")
+st.title(":speech_balloon: Assistant API Chat")
+
+st.sidebar.caption(f"**Session ID**:  \n  {st.session_state.session_id}")
+st.sidebar.header("Configuration")
+thread_id = st.sidebar.text_input("Enter your Thread ID:")
+    
+if st.sidebar.button("Start Chat"):
+
+    st.session_state.start_chat = True
+    if thread_id:
+        st.session_state.thread_id = thread_id
+    else:
+        thread = client.beta.threads.create(
+            metadata={
+                'session_id': st.session_state.session_id,
+            }
+        )
+        st.session_state.thread_id = thread.id
+
+    st.caption(f"**Thread ID**: {st.session_state.thread_id}")
+    st.session_state.messages = client.beta.threads.messages.list(
+        thread_id=st.session_state.thread_id
+    )
+
+#def process_message(message):
+#    message_content = message.content[0].text
+#    annotations = message_content.annotations if hasattr(message_content, 'annotations') else []
+#    citations = []
+#    full_response = message_content.value + '\n\n' + '\n'.join(citations)
+#    return full_response
+
+if st.session_state.start_chat:
+
+    for message in reversed(st.session_state.messages.data):
+        with st.chat_message(message.role):
+            st.markdown(message.content[0].text.value)
+
+    if prompt := st.chat_input():
+
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        client.beta.threads.messages.create(
+            thread_id=st.session_state.thread_id,
+            role="user",
+            content=prompt
+        )
+
+        run = client.beta.threads.runs.create(
+            thread_id=st.session_state.thread_id,
+            assistant_id=assistant_id,
+        )
+
+        while run.status != 'completed':
+            time.sleep(1)
+            run = client.beta.threads.runs.retrieve(
+                thread_id=st.session_state.thread_id,
+                run_id=run.id
+            )
+
+        st.session_state.messages = client.beta.threads.messages.list(
+            thread_id=st.session_state.thread_id
+        )
+
+        for message in reversed(st.session_state.messages.data):
+            if message.run_id == run.id and message.role == "assistant":
+                with st.chat_message("assistant"):
+                    st.markdown(message.content[0].text.value)
+else:
+    st.write("Click on 'Start Chat' to start/continue a thread.")

requirements.txt

@@ -0,0 +1,2 @@
+streamlit==1.28.2
+openai==1.2.4