Thang commited on
Commit
d015c2d
·
1 Parent(s): 494e8a0

Add application file

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ # from inference import *
3
+
4
+
5
+ st.title("💬 Chatbot")
6
+ st.caption("🚀 A streamlit chatbot powered by Mistral LLM")
7
+
8
+ if "messages" not in st.session_state:
9
+ st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
10
+
11
+ for msg in st.session_state.messages:
12
+ st.chat_message(msg["role"]).write(msg["content"])
13
+
14
+ if prompt := st.chat_input():
15
+ # client = OpenAI(api_key=openai_api_key)
16
+ st.session_state.messages.append({"role": "user", "content": prompt})
17
+ st.chat_message("user").write(prompt)
18
+
19
+ response = "" #client.chat.completions.create(model="gpt-3.5-turbo", messages=st.session_state.messages)
20
+ msg = "" #response.choices[0].message.content
21
+
22
+ st.session_state.messages.append({"role": "assistant", "content": msg})
23
+ st.chat_message("assistant").write(msg)
24
+