Update app.py

app.py

@@ -9,6 +9,34 @@ os.system('pip install dgl==1.0.2+cu116 -f https://data.dgl.ai/wheels/cu116/repo
 os.environ["DGLBACKEND"] = "pytorch"
 print('Modules installed')
 
+# 여기에 args 초기화 코드 추가 --------------------------------
+# 기본 args 설정
+if not os.path.exists('./tmp'):
+    os.makedirs('./tmp')
+
+if not os.path.exists('./tmp/args.json'):
+    default_args = {
+        'checkpoint': None,
+        'dump_trb': False,
+        'dump_args': True,
+        'save_best_plddt': True,
+        'T': 25,
+        'strand_bias': 0.0,
+        'loop_bias': 0.0,
+        'helix_bias': 0.0,
+        'd_t1d': 24,
+        'potentials': None,
+        'potential_scale': None,
+        'aa_composition': None
+    }
+    with open('./tmp/args.json', 'w') as f:
+        json.dump(default_args, f)
+
+# args 로드
+with open('./tmp/args.json', 'r') as f:
+    args = json.load(f)
+
+
 # 필수 라이브러리 임포트
 from datasets import load_dataset
 import plotly.graph_objects as go
@@ -583,20 +611,23 @@ def combined_generation(name, strength, flexibility, speed, defense, size, abili
             None
         )
 
-# Gradio 인터페이스 수정
 with gr.Blocks(theme='ParityError/Interstellar') as demo:
     with gr.Row():
-        # 왼쪽 열: 챗봇 및 컨트롤 패널
         with gr.Column(scale=1):
             # 챗봇 인터페이스
             gr.Markdown("# 🤖 AI 단백질 설계 도우미")
             chatbot = gr.Chatbot(height=600)
-            msg = gr.Textbox(
-                label="메시지를 입력하세요",
-                placeholder="예: COVID-19를 치료할 수 있는 단백질을 생성해주세요",
-                lines=2
-            )
+            # 여기에 수정된 입력 필드 추가 --------------------------------
+            with gr.Row():
+                msg = gr.Textbox(
+                    label="메시지를 입력하세요",
+                    placeholder="예: COVID-19를 치료할 수 있는 단백질을 생성해주세요",
+                    lines=2,
+                    scale=4
+                )
+                submit_btn = gr.Button("전송", variant="primary", scale=1)
             clear = gr.Button("대화 내용 지우기")
+
             
             with gr.Accordion("채팅 설정", open=False):
                 system_message = gr.Textbox(
@@ -886,7 +917,6 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
     msg.submit(process_chat, [msg, chatbot], [chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
-    # UI 컨트롤 이벤트
     seq_opt.change(
         fn=toggle_seq_input,
         inputs=[seq_opt],
@@ -894,6 +924,8 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         queue=False
     )
 
+
+
     sec_str_opt.change(
         fn=toggle_secondary_structure,
         inputs=[sec_str_opt],
@@ -949,12 +981,6 @@ 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(
@@ -964,6 +990,14 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
     )
 
 
+    submit_btn.click(respond, 
+                    [msg, chatbot, system_message, max_tokens, temperature, top_p], 
+                    [chatbot])
+    msg.submit(respond, 
+              [msg, chatbot, system_message, max_tokens, temperature, top_p], 
+              [chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
+
 # 실행
 demo.queue()
 demo.launch(debug=True)