Spaces:
Runtime error
Runtime error
Tim F
commited on
Commit
·
e16892f
1
Parent(s):
77e791d
added secrets file for openai key
Browse files- .streamlit/secrets.toml +1 -0
- app.py +7 -5
.streamlit/secrets.toml
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
OPENAI_API_KEY = "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"
|
app.py
CHANGED
|
@@ -12,11 +12,11 @@ from langchain_core.prompts import ChatPromptTemplate
|
|
| 12 |
|
| 13 |
from langchain_community.document_loaders import UnstructuredFileLoader
|
| 14 |
|
| 15 |
-
|
| 16 |
-
os.environ["OPENAI_API_KEY"] = "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"
|
| 17 |
|
| 18 |
# Get all the filenames from the docs folder
|
| 19 |
files = glob.glob("./docs/*.txt")
|
|
|
|
| 20 |
# Load files into readable documents
|
| 21 |
docs = []
|
| 22 |
for file in files:
|
|
@@ -68,7 +68,7 @@ rag_chain = (
|
|
| 68 |
| StrOutputParser()
|
| 69 |
)
|
| 70 |
|
| 71 |
-
st.title("
|
| 72 |
|
| 73 |
if "messages" not in st.session_state:
|
| 74 |
st.session_state["messages"] = [{"role": "assistant", "content": "Ahoi! Ich bin Volker. Wie kann ich dir helfen?"}]
|
|
@@ -78,5 +78,7 @@ for msg in st.session_state.messages:
|
|
| 78 |
|
| 79 |
if prompt := st.chat_input():
|
| 80 |
st.chat_message("user").write(prompt)
|
| 81 |
-
|
| 82 |
-
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
from langchain_community.document_loaders import UnstructuredFileLoader
|
| 14 |
|
| 15 |
+
OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
|
|
|
|
| 16 |
|
| 17 |
# Get all the filenames from the docs folder
|
| 18 |
files = glob.glob("./docs/*.txt")
|
| 19 |
+
|
| 20 |
# Load files into readable documents
|
| 21 |
docs = []
|
| 22 |
for file in files:
|
|
|
|
| 68 |
| StrOutputParser()
|
| 69 |
)
|
| 70 |
|
| 71 |
+
st.title("🐔 Volker-Chat")
|
| 72 |
|
| 73 |
if "messages" not in st.session_state:
|
| 74 |
st.session_state["messages"] = [{"role": "assistant", "content": "Ahoi! Ich bin Volker. Wie kann ich dir helfen?"}]
|
|
|
|
| 78 |
|
| 79 |
if prompt := st.chat_input():
|
| 80 |
st.chat_message("user").write(prompt)
|
| 81 |
+
with get_openai_callback() as cb:
|
| 82 |
+
response = rag_chain.invoke(prompt)
|
| 83 |
+
st.chat_message("assistant").write(response)
|
| 84 |
+
st.chat_message("assistant").write(cb)
|