Kims12 commited on
Commit
95b8a75
·
verified ·
1 Parent(s): 6d1479a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -9
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
  import openai
@@ -198,7 +199,6 @@ def claude_respond(
198
 
199
  # Anthropic Messages API에 맞게 "system" role은 top-level 파라미터로,
200
  # 나머지는 user/assistant만 messages에 넣음
201
- # (messages 내 "system" role이 있으면 오류 발생)
202
  anthro_messages = []
203
  for human, assistant in chat_history:
204
  if human:
@@ -340,6 +340,14 @@ with gr.Blocks() as demo:
340
  )
341
  chatgpt_clear_button.click(clear_conversation, outputs=chatgpt_chatbot, queue=False)
342
 
 
 
 
 
 
 
 
 
343
  with gr.Tab("Claude"):
344
  with gr.Row():
345
  claude_system_message = gr.Textbox(
@@ -350,23 +358,24 @@ with gr.Blocks() as demo:
350
  label="System Message",
351
  lines=3
352
  )
353
- claude_max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens")
354
- claude_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
 
 
 
 
355
  claude_top_p = gr.Slider(
356
- minimum=0.1,
357
- maximum=1.0,
358
- value=0.95,
359
- step=0.05,
360
- label="Top-P",
361
  )
362
 
363
  claude_chatbot = gr.Chatbot(height=600)
364
  claude_msg = gr.Textbox(label="메세지를 입력하세요")
 
365
  with gr.Row():
366
  claude_submit_button = gr.Button("전송")
367
  claude_clear_button = gr.Button("대화 내역 지우기")
368
 
369
- # Claude 전용 함수 호출
370
  claude_msg.submit(
371
  claude_respond,
372
  [claude_msg, claude_chatbot, claude_system_message, claude_max_tokens, claude_temperature, claude_top_p],
@@ -379,5 +388,7 @@ with gr.Blocks() as demo:
379
  )
380
  claude_clear_button.click(clear_conversation, outputs=claude_chatbot, queue=False)
381
 
 
 
382
  if __name__ == "__main__":
383
  demo.launch()
 
1
+ ###################### (원본코드 시작) ######################
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
  import openai
 
199
 
200
  # Anthropic Messages API에 맞게 "system" role은 top-level 파라미터로,
201
  # 나머지는 user/assistant만 messages에 넣음
 
202
  anthro_messages = []
203
  for human, assistant in chat_history:
204
  if human:
 
340
  )
341
  chatgpt_clear_button.click(clear_conversation, outputs=chatgpt_chatbot, queue=False)
342
 
343
+ ###################### (원본코드 끝) ######################
344
+
345
+
346
+ # ------------------------- 여기부터 "클로드 전용 탭" 추가 -------------------------
347
+ # - 원본코드에 이미 claude_respond 함수가 있으므로 그대로 활용
348
+ # - 탭만 추가하여 UI 구성
349
+
350
+ with demo: # 원본코드에서 "with gr.Blocks() as demo:" 를 이미 열었음
351
  with gr.Tab("Claude"):
352
  with gr.Row():
353
  claude_system_message = gr.Textbox(
 
358
  label="System Message",
359
  lines=3
360
  )
361
+ claude_max_tokens = gr.Slider(
362
+ minimum=1, maximum=2048, value=512, step=1, label="Max Tokens"
363
+ )
364
+ claude_temperature = gr.Slider(
365
+ minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature"
366
+ )
367
  claude_top_p = gr.Slider(
368
+ minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P",
 
 
 
 
369
  )
370
 
371
  claude_chatbot = gr.Chatbot(height=600)
372
  claude_msg = gr.Textbox(label="메세지를 입력하세요")
373
+
374
  with gr.Row():
375
  claude_submit_button = gr.Button("전송")
376
  claude_clear_button = gr.Button("대화 내역 지우기")
377
 
378
+ # 이미 정의된 claude_respond 함수를 그대로 사용
379
  claude_msg.submit(
380
  claude_respond,
381
  [claude_msg, claude_chatbot, claude_system_message, claude_max_tokens, claude_temperature, claude_top_p],
 
388
  )
389
  claude_clear_button.click(clear_conversation, outputs=claude_chatbot, queue=False)
390
 
391
+
392
+ # 원본 코드의 데모 런치 부분도 수정 금지
393
  if __name__ == "__main__":
394
  demo.launch()