anpigon commited on
Commit
3f5aab0
1 Parent(s): 5399754

style(app.py): remove verbose flag from RetrievalQAWithSourcesChain initialization to reduce console output

Browse files

refactor(app.py): remove commented out code for displaying source documents
refactor(app.py): remove unused avatar image file and related code
refactor(app.py): simplify gr.Blocks() initialization and remove debug flag

Files changed (2) hide show
  1. app.py +12 -33
  2. images/avatar.webp +0 -0
app.py CHANGED
@@ -42,51 +42,30 @@ chain = RetrievalQAWithSourcesChain.from_chain_type(
42
  return_source_documents=True,
43
  chain_type_kwargs=chain_type_kwargs,
44
  reduce_k_below_max_tokens=True,
45
- verbose=True,
46
  )
47
 
48
 
49
- # 채팅봇의 응답을 처리하는 함수를 정의합니다.
50
  def respond(message, chat_history):
51
  if len(chat_history) == 0:
52
  initial_greeting = "안녕하세요!\n저는 데일 카네기처럼 경험과 지식을 갖춘 인공지능 ChatGPT입니다. 데일 카네기는 인간관계와 소통에 대한 전문가이며, 많은 사람들이 그의 조언을 참고하고 있습니다. 어떤 도움이 필요하신가요? 데일 카네기와 관련된 질문이 있으시면 편안하게 물어보세요!"
53
  chat_history.append(("Bot", initial_greeting))
54
-
55
  result = chain(message)
56
 
57
  bot_message = result["answer"]
58
- source_documents = result["source_documents"]
59
-
60
- # 문서 출처를 이쁘게 출력합니다.
61
- # if source_documents:
62
- # bot_message += "\n\n**문서 출처:**\n"
63
- # for doc in source_documents:
64
- # source = doc.metadata['source']
65
- # page = doc.metadata['page']
66
- # bot_message += f"1. {source}({page})\n"
67
-
68
- # 채팅 기록에 사용자의 메시지와 봇의 응답을 추가합니다.
69
- chat_history.append((message, bot_message))
70
-
71
- # 수정된 채팅 기록을 반환합니다.
72
  return "", chat_history
73
 
74
 
75
- # gr.Blocks() 사용하여 인터페이스를 생성합니다.
76
- with gr.Blocks(theme=gr.themes.Glass()) as demo:
77
- gr.Image("images/avatar.webp", label="데일 카네기", height=156, width=104)
78
  gr.Markdown("# 안녕하세요. 데일 카네기와 대화해보세요.")
79
  initial_greeting = "안녕하세요!\n저는 데일 카네기처럼 경험과 지식을 갖춘 인공지능 ChatGPT입니다. 데일 카네기는 인간관계와 소통에 대한 전문가이며, 많은 사람들이 그의 조언을 참고하고 있습니다. 어떤 도움이 필요하신가요? 데일 카네기와 관련된 질문이 있으시면 편안하게 물어보세요!"
80
- chatbot = gr.Chatbot(label="채팅창", value=[(None, initial_greeting)]) # '채팅창'이라는 레이블을 가진 채팅봇 컴포넌트를 생성합니다.
81
- msg = gr.Textbox(label="입력") # '입력'이라는 레이블을 가진 텍스트박스를 생성합니다.
82
- clear = gr.Button("초기화") # '초기화'라는 레이블을 가진 버튼을 생성합니다.
83
- msg.submit(
84
- respond, [msg, chatbot], [msg, chatbot]
85
- ) # 텍스트박스에 메시지를 입력하고 제출하면 respond 함수가 호출되도록 합니다.
86
- clear.click(
87
- lambda: None, None, chatbot, queue=False
88
- ) # '초기화' 버튼을 클릭하면 채팅 기록을 초기화합니다.
89
-
90
- demo.launch(
91
- debug=True
92
- ) # 인터페이스를 실행합니다. 실행하면 사용자는 '입력' 텍스트박스에 메시지를 작성하고 제출할 수 있으며, '초기화' 버튼을 통해 채팅 기록을 초기화 할 수 있습니다.
 
42
  return_source_documents=True,
43
  chain_type_kwargs=chain_type_kwargs,
44
  reduce_k_below_max_tokens=True,
45
+ verbose=False,
46
  )
47
 
48
 
 
49
  def respond(message, chat_history):
50
  if len(chat_history) == 0:
51
  initial_greeting = "안녕하세요!\n저는 데일 카네기처럼 경험과 지식을 갖춘 인공지능 ChatGPT입니다. 데일 카네기는 인간관계와 소통에 대한 전문가이며, 많은 사람들이 그의 조언을 참고하고 있습니다. 어떤 도움이 필요하신가요? 데일 카네기와 관련된 질문이 있으시면 편안하게 물어보세요!"
52
  chat_history.append(("Bot", initial_greeting))
53
+
54
  result = chain(message)
55
 
56
  bot_message = result["answer"]
57
+ chat_history.append((message, bot_message))
58
+
 
 
 
 
 
 
 
 
 
 
 
 
59
  return "", chat_history
60
 
61
 
62
+ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
 
63
  gr.Markdown("# 안녕하세요. 데일 카네기와 대화해보세요.")
64
  initial_greeting = "안녕하세요!\n저는 데일 카네기처럼 경험과 지식을 갖춘 인공지능 ChatGPT입니다. 데일 카네기는 인간관계와 소통에 대한 전문가이며, 많은 사람들이 그의 조언을 참고하고 있습니다. 어떤 도움이 필요하신가요? 데일 카네기와 관련된 질문이 있으시면 편안하게 물어보세요!"
65
+ chatbot = gr.Chatbot(label="채팅창", value=[(None, initial_greeting)])
66
+ msg = gr.Textbox(label="입력")
67
+ clear = gr.Button("초기화")
68
+ msg.submit(respond, [msg, chatbot], [msg, chatbot])
69
+ clear.click(lambda: None, None, chatbot, queue=False)
70
+
71
+ demo.launch(debug=False)
 
 
 
 
 
 
images/avatar.webp DELETED
Binary file (15.8 kB)