Update app.py

app.py

@@ -1121,76 +1121,77 @@ try:
 except Exception as e:
     print(f"Warning: Could not download checkpoint files: {str(e)}")
 
+
+
 with gr.Blocks(theme='ParityError/Interstellar') as demo:
+    # 예제 프롬프트 리스트 정의
+    example_prompts = [
+        "암세포만 선택적으로 공격하는 면역 단백질을 설계해주세요",
+        "COVID-19 스파이크 단백질에 강하게 결합하는 항체 단백질을 생성해주세요",
+        "혈당 수준을 실시간으로 모니터링하는 단백질을 만들어주세요",
+        "알츠하이머 베타 아밀로이드를 분해할 수 있는 효소 단백질을 설계해주세요",
+        "암 마커를 초고감도로 검출하는 단백질을 생성해주세요",
+        "플라스틱을 분해할 수 있는 효소 단백질을 설계해주세요",
+        "이산화탄소를 효율적으로 포집하는 단백질을 만들어주세요",
+        "식품의 보존기간을 연장하는 항균 단백질을 설계해주세요",
+        "수소 생산을 촉진하는 효소 단백질을 생성해주세요",
+        "혈전을 효과적으로 분해하는 단백질을 설계해주세요",
+        "인슐린 저항성을 개선하는 새로운 호르몬 단백질을 만들어주세요",
+        "식물의 가뭄 저항성을 높이는 단백질을 설계해주세요",
+        "태양광 에너지를 효율적으로 포집하는 단백질을 생성해주세요",
+        "해수에서 중금속을 제거하는 단백질을 설계해주세요",
+        "질소 고정 효율을 높이는 효소 단백질을 만들어주세요",
+        "신경전달물질을 감지하는 나노바이오센서를 설계해주세요",
+        "특정 독성 물질을 감지하는 바이오센서 단백질을 생성해주세요",
+        "배터리 성능을 향상시키는 전도성 단백질을 설계해주세요",
+        "극한의 고온에서도 안정한 초내열성 단백질을 만들어주세요",
+        "생분해성 플라스틱 생산을 위한 중합효소를 설계해주세요",
+        "식물의 병해충 저항성을 강화하는 단백질을 생성해주세요",
+        "과일의 숙성을 조절하는 단백질을 설계해주세요",
+        "대기 오염물질을 실시간 감지하는 단백질을 만들어주세요",
+        "석유 오염을 분해하는 해양 단백질을 설계해주세요",
+        "극저온에서 활성을 유지하는 저온 적응 단백질을 생성해주세요",
+        "고압 환경에서 안정한 심해 적응 단백질을 설계해주세요",
+        "강한 산성 환경에서 작동하는 단백질을 만들어주세요",
+        "나노 구조체를 자가 조립하는 단백질을 설계해주세요",
+        "방사능 물질을 안전하게 분해하는 단백질을 생성해주세요",
+        "방사선에 저항성이 있는 우주 환경용 단백질을 설계해주세요"
+    ]
+
     with gr.Row():
         with gr.Column(scale=1):
             # 챗봇 인터페이스
-            gr.Markdown("# 🤖 AI 단백질 설계 도우미")
-            # 여기를 수정
+            gr.Markdown("# 🤖 ProteinGPT: AI 단백질 생성기기")
+            
+            # 예제 프롬프트 드롭다운과 버튼을 Row로 배치
+            with gr.Row():
+                example_dropdown = gr.Dropdown(
+                    choices=example_prompts,
+                    label="예제 프롬프트 선택",
+                    info="원하는 예제를 선택하세요",
+                    scale=4
+                )
+                example_btn = gr.Button("👉 예제 사용", scale=1)
+            
+            # 챗봇
             chatbot = gr.Chatbot(
                 height=600,
-                type='messages'  # 메시지 형식 지정
+                type='messages'
             )
-
-    # 예제 프롬프트 드롭다운 추가
-            example_prompts = [
-                "암세포만 선택적으로 공격하는 면역 단백질을 설계해주세요",
-                "COVID-19 스파이크 단백질에 강하게 결합하는 항체 단백질을 생성해주세요",
-                "혈당 수준을 실시간으로 모니터링하는 단백질을 만들어주세요",
-                "알츠하이머 베타 아밀로이드를 분해할 수 있는 효소 단백질을 설계해주세요",
-                "암 마커를 초고감도로 검출하는 단백질을 생성해주세요",
-                "플라스틱을 분해할 수 있는 효소 단백질을 설계해주세요",
-                "이산화탄소를 효율적으로 포집하는 단백질을 만들어주세요",
-                "식품의 보존기간을 연장하는 항균 단백질을 설계해주세요",
-                "수소 생산을 촉진하는 효소 단백질을 생성해주세요",
-                "혈전을 효과적으로 분해하는 단백질을 설계해주세요",
-                "인슐린 저항성을 개선하는 새로운 호르몬 단백질을 만들어주세요",
-                "식물의 가뭄 저항성을 높이는 단백질을 설계해주세요",
-                "태양광 에너지를 효율적으로 포집하는 단백질을 생성해주세요",
-                "해수에서 중금속을 제거하는 단백질을 설계해주세요",
-                "질소 고정 효율을 높이는 효소 단백질을 만들어주세요",
-                "신경전달물질을 감지하는 나노바이오센서를 설계해주세요",
-                "특정 독성 물질을 감지하는 바이오센서 단백질을 생성해주세요",
-                "배터리 성능을 향상시키는 전도성 단백질을 설계해주세요",
-                "극한의 고온에서도 안정한 초내열성 단백질을 만들어주세요",
-                "생분해성 플라스틱 생산을 위한 중합효소를 설계해주세요",
-                "식물의 병해충 저항성을 강화하는 단백질을 생성해주세요",
-                "과일의 숙성을 조절하는 단백질을 설계해주세요",
-                "대기 오염물질을 실시간 감지하는 단백질을 만들어주세요",
-                "석유 오염을 분해하는 해양 단백질을 설계해주세요",
-                "극저온에서 활성을 유지하는 저온 적응 단백질을 생성해주세요",
-                "고압 환경에서 안정한 심해 적응 단백질을 설계해주세요",
-                "강한 산성 환경에서 작동하는 단백질을 만들어주세요",
-                "나노 구조체를 자가 조립하는 단백질을 설계해주세요",
-                "방사능 물질을 안전하게 분해하는 단백질을 생성해주세요",
-                "방사선에 저항성이 있는 우주 환경용 단백질을 설계해주세요"
-            ]
-
-    
-    
-    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("대화 내용 지우기")
-
-# 예제 선택 시 입력창에 자동 입력
-def use_example(example):
-    return example
-
-use_example_btn.click(
-    fn=use_example,
-    inputs=[example_dropdown],
-    outputs=[msg]
-)
+            
+            # 메시지 입력
+            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(
                     value="당신은 단백질 설계를 도와주는 전문가입니다.",
@@ -1218,16 +1219,15 @@ use_example_btn.click(
                     label="Top-P"
                 )
 
-
             # 탭 인터페이스
             with gr.Tabs():
-                with gr.TabItem("🦸‍♂️ 커스텀 디자인"):
+                with gr.TabItem("🧬 커스텀 디자인"):
                     gr.Markdown("""
                     ### ✨ 당신만의 특별한 커스텀을 만들어보세요!
                     각 능력치를 조절하면 커스텀된 단백질이 자동으로 설계됩니다.
                     """)
                     
-                    # 히어로 기본 정보
+                    # 기본 정보
                     hero_name = gr.Textbox(
                         label="커스텀 이름", 
                         placeholder="당신의 커스텀 단백질 이름을 지어주세요!",
@@ -1264,7 +1264,7 @@ use_example_btn.click(
                             info="음수: 수중 활동에 특화, 양수: 지상 활동에 특화"
                         )
                     
-                    # 히어로 크기 설정
+                    # 크기 설정
                     hero_size = gr.Slider(
                         minimum=50, maximum=200, 
                         label="📏 커스텀 단백질 크기", 
@@ -1275,7 +1275,7 @@ use_example_btn.click(
                     # 특수 능력 설정
                     with gr.Accordion("🌟 특수 능력", open=False):
                         gr.Markdown("""
-                        특수 능력을 선택하면 커스텀 단백질질에 특별한 구조가 추가됩니다.
+                        특수 능력을 선택하면 커스텀 단백질에 특별한 구조가 추가됩니다.
                         - 자가 회복: 단백질 구조 복구 능력 강화
                         - 원거리 공격: 특수한 구조적 돌출부 형성
                         - 방어막 생성: 안정적인 보호층 구조 ���성
@@ -1460,7 +1460,7 @@ use_example_btn.click(
 
         # 오른쪽 열: 결과 표시
         with gr.Column(scale=1):
-            gr.Markdown("## 🦸‍♂️ 커스텀 단백질 프로필")
+            gr.Markdown("## 🧬 커스텀 단백질 프로필")
             hero_stats = gr.Plot(label="능력치 분석")
             hero_description = gr.Textbox(label="커스텀 단백질 특성", lines=3)
             
@@ -1474,11 +1474,26 @@ use_example_btn.click(
             gr.Markdown("#### 🔬 커스텀 단백질 구조")
             output_viewer = gr.HTML()
 
-    # 이벤트 연결
+    # 예제 선택 이벤트 연결
+    def use_example(example):
+        if example:
+            return example
+        return ""
+
+    example_btn.click(
+        fn=use_example,
+        inputs=[example_dropdown],
+        outputs=[msg]
+    )            
+
+
+
+# 이벤트 연결
     # 챗봇 이벤트
     msg.submit(process_chat, [msg, chatbot], [chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
+    # 입력 방식 변경 이벤트
     seq_opt.change(
         fn=toggle_seq_input,
         inputs=[seq_opt],
@@ -1486,8 +1501,7 @@ use_example_btn.click(
         queue=False
     )
 
-
-
+    # 구조 설정 방식 변경 이벤트
     sec_str_opt.change(
         fn=toggle_secondary_structure,
         inputs=[sec_str_opt],
@@ -1495,12 +1509,14 @@ use_example_btn.click(
         queue=False
     )
 
+    # 미리보기 이벤트
     preview_btn.click(
         get_motif_preview,
         inputs=[pdb_id_code, contigs],
         outputs=[preview_viewer, rewrite_pdb]
     )
 
+    # PSSM 분석 이벤트
     pssm_gen_btn.click(
         get_pssm,
         inputs=[fasta_msa, input_pssm],
@@ -1572,7 +1588,7 @@ use_example_btn.click(
         ]
     )
 
-# 채팅 내용 지우기 버튼
+    # 채팅 내용 지우기
     clear.click(
         lambda: (None, None, None, None, None, None, None),
         None,
@@ -1580,7 +1596,6 @@ use_example_btn.click(
         queue=False
     )
 
-
     # 챗봇 응답에 따른 결과 업데이트
     msg.submit(
         update_protein_display,
@@ -1588,7 +1603,7 @@ use_example_btn.click(
         outputs=[hero_stats, hero_description, output_seq, output_pdb, output_viewer, plddt_plot]
     )
 
-
+    # 기본 챗봇 응답 처리
     submit_btn.click(respond, 
                     [msg, chatbot, system_message, max_tokens, temperature, top_p], 
                     [chatbot])
@@ -1597,8 +1612,7 @@ use_example_btn.click(
               [chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
-
-# 이벤트 핸들러 연결
+    # 진행 상태 표시 이벤트
     msg.submit(
         fn=process_chat_and_generate,
         inputs=[msg, chatbot],
@@ -1629,7 +1643,6 @@ use_example_btn.click(
         show_progress=True
     )
 
-
 # 실행
 demo.queue()
-demo.launch(debug=True)
+demo.launch(debug=True)