Update app.py

app.py

@@ -583,6 +583,7 @@ def combined_generation(name, strength, flexibility, speed, defense, size, abili
             None
         )
 
+# Gradio 인터페이스 수정
 with gr.Blocks(theme='ParityError/Interstellar') as demo:
     with gr.Row():
         # 왼쪽 열: 챗봇 및 컨트롤 패널
@@ -590,6 +591,12 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             # 챗봇 인터페이스
             gr.Markdown("# 🤖 AI 단백질 설계 도우미")
             chatbot = gr.Chatbot(height=600)
+            msg = gr.Textbox(
+                label="메시지를 입력하세요",
+                placeholder="예: COVID-19를 치료할 수 있는 단백질을 생성해주세요",
+                lines=2
+            )
+            clear = gr.Button("대화 내용 지우기")
             
             with gr.Accordion("채팅 설정", open=False):
                 system_message = gr.Textbox(
@@ -942,6 +949,13 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             ]
         )
 
+# 이벤트 연결
+    # 챗봇 이벤트
+    msg.submit(respond, 
+              [msg, chatbot, system_message, max_tokens, temperature, top_p], 
+              [chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
+
     # 챗봇 응답에 따른 결과 업데이트
     msg.submit(
         update_protein_display,
@@ -949,16 +963,6 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         outputs=[hero_stats, hero_description, output_seq, output_pdb, output_viewer, plddt_plot]
     )
 
-    chat_interface = gr.ChatInterface(
-        respond,
-        additional_inputs=[
-            system_message,
-            max_tokens,
-            temperature,
-            top_p,
-        ],
-        chatbot=chatbot,
-    )
 
 # 실행
 demo.queue()