Update app.py

app.py

@@ -12,19 +12,6 @@ from langchain.document_loaders import TextLoader
 from langchain.document_loaders import DirectoryLoader
 
 
-# from streamlit_chat import message
-
-# 스트림릿이랑 파이썬, vs코드 같은 폴더에
-# 이 파일도 같은 폴더에 넣어야하는지는 모르겠움,,
-
-# import streamlit as st
-# from io import StringIO
-
-# 경로 /Users/gyuribyun/Documents/legoapp
-# 경로 터미널에 입력 후(cd 경로 복붙)
-# streamlit run app.py로 링크 받기
-
-
 # 타이틀 적용, # 특수 이모티콘 삽입 예시
 # emoji: https://streamlit-emoji-shortcodes-streamlit-app-gwckff.streamlit.app/
 st.title(':robot_face:한국사봇? 너 정말 똑똑하니?')
@@ -38,10 +25,12 @@ st.markdown('###### 질문, 요약 등 다양한 부탁을 해 보세요! 교과
 
 
 api_key = st.text_input(label='OpenAI API 키를 입력하세요', type='password')
+OpenAI.api_key = api_key
+
 
 if api_key:
     # OpenAI API를 사용하기 위한 처리 과정을 함수로 정의
-    def initialize_openai_processing(api_key):
+    def initialize_openai_processing():
         
         loader = DirectoryLoader('./khistory_data', glob="*.txt", loader_cls=TextLoader)
         documents = loader.load()
@@ -49,8 +38,8 @@ if api_key:
         texts = text_splitter.split_documents(documents)
         
         persist_directory = 'db'
-        #embedding = OpenAIEmbeddings()
-        embedding = OpenAIEmbeddings(api_key=api_key)  # API 키를 생성자에 전달
+        embedding = OpenAIEmbeddings()
+        #embedding = OpenAIEmbeddings(api_key=api_key)  # API 키를 생성자에 전달
         vectordb = Chroma.from_documents(
             documents=texts,
             embedding=embedding,