Update app.py

app.py

@@ -65,6 +65,8 @@ def respond(message, chat_history):
         # for i, doc in enumerate(result['source_documents']):
         #     bot_message += str(i+1) + '. ' + doc.metadata['source'] + ' '
 
+        historySave(message=message, answer=str(result['result']).replace("'",""))
+        
         # 채팅 기록에 사용자의 메시지와 봇의 응답을 추가.
         chat_history.append((message, bot_message))
         
@@ -74,6 +76,21 @@ def respond(message, chat_history):
         
         return " ", chat_history
 
+def historySave(message, answer):
+
+    conn = pymssql.connect(host=r"(127.0.0.1)", database='Chatbot_Manage', charset='utf8')
+    conn.autocommit(True) # 오토 커밋 활성화
+    # Connection 으로부터 Cursor 생성
+    cursor = conn.cursor()
+
+    SystemType = "OpenAI(Real LLM)"
+    
+    # SQL문 실행'
+    _sql = "EXEC ChatHistory_InsUpd '" + SystemType + "','" + message + "', '" + answer + "'"
+    cursor.execute(_sql)
+    
+    conn.close()    ## 연결 끊기
+    
 import gradio as gr
 
 # 챗봇 설명