Upload app.py

app.py

@@ -24,7 +24,7 @@ def LoadData(openai_key):
         
         persist_directory = 'realdb_LLM'
 
-        embedding = OpenAIEmbeddings()
+        embedding = OpenAIEmbeddings(openai_key)
 
         vectordb = Chroma(
             persist_directory=persist_directory, 
@@ -38,15 +38,9 @@ def LoadData(openai_key):
     else:
         return "사용하시는 API Key를 입력하여 주시기 바랍니다."
 
-def process_llm_response(llm_response):
-    print(llm_response['result'])
-    print('\n\nSources:')
-    for source in llm_response["source_documents"]:
-        print(source.metadata['source'])
-        
-        
+
 # 챗봇의 답변을 처리하는 함수
-def respond(message, chat_history, temperature):
+def respond(message, chat_history):
     try:
         qa_chain = RetrievalQA.from_chain_type(
             llm=OpenAI(temperature=0.4),
@@ -75,9 +69,6 @@ def respond(message, chat_history, temperature):
         
         return " ", chat_history
 
-        # return "", chat_history
-
-    
 import gradio as gr
 
 # 챗봇 설명