Update app.py

app.py

@@ -544,104 +544,190 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             gr.Markdown("# 🦸‍♂️ 슈퍼히어로 단백질 만들기")
             
             with gr.Tabs():
-                with gr.TabItem("히어로 디자인"):
+                with gr.TabItem("🦸‍♂️ 히어로 디자인"):
+                    gr.Markdown("""
+                    ### ✨ 당신만의 특별한 히어로를 만들어보세요!
+                    각 능력치를 조절하면 히어로의 DNA가 자동으로 설계됩니다.
+                    """)
+                    
                     # 히어로 기본 정보
-                    hero_name = gr.Textbox(label="히어로 이름", placeholder="당신의 히어로 이름을 지어주세요!")
+                    hero_name = gr.Textbox(
+                        label="히어로 이름", 
+                        placeholder="당신의 히어로 이름을 지어주세요!",
+                        info="히어로의 정체성을 나타내는 이름을 입력하세요"
+                    )
                     
                     # 능력치 설정
                     gr.Markdown("### 💪 히어로 능력치 설정")
                     with gr.Row():
-                        strength = gr.Slider(minimum=0.0, maximum=0.05, label="💪 초강력(근력)", value=0.02)
-                        flexibility = gr.Slider(minimum=0.0, maximum=0.05, label="🤸‍♂️ 유연성", value=0.02)
+                        strength = gr.Slider(
+                            minimum=0.0, maximum=0.05, 
+                            label="💪 초강력(근력)", 
+                            value=0.02,
+                            info="단단한 베타시트 구조로 강력한 힘을 생성합니다"
+                        )
+                        flexibility = gr.Slider(
+                            minimum=0.0, maximum=0.05, 
+                            label="🤸‍♂️ 유연성", 
+                            value=0.02,
+                            info="나선형 알파헬릭스 구조로 유연한 움직임을 가능하게 합니다"
+                        )
                     
                     with gr.Row():
-                        speed = gr.Slider(minimum=0.0, maximum=0.20, label="⚡ 스피드", value=0.1)
-                        defense = gr.Slider(minimum=-10, maximum=10, label="🛡️ 방어력", value=0)
+                        speed = gr.Slider(
+                            minimum=0.0, maximum=0.20, 
+                            label="⚡ 스피드", 
+                            value=0.1,
+                            info="루프 구조로 빠른 움직임을 구현합니다"
+                        )
+                        defense = gr.Slider(
+                            minimum=-10, maximum=10, 
+                            label="🛡️ 방어력", 
+                            value=0,
+                            info="음수: 수중 활동에 특화, 양수: 지상 활동에 특화"
+                        )
                     
                     # 히어로 크기 설정
-                    hero_size = gr.Slider(minimum=50, maximum=200, label="히어로 크기", value=100)
+                    hero_size = gr.Slider(
+                        minimum=50, maximum=200, 
+                        label="📏 히어로 크기", 
+                        value=100,
+                        info="히어로의 전체적인 크기를 결정합니다"
+                    )
                     
                     # 특수 능력 설정
                     with gr.Accordion("🌟 특수 능력", open=False):
+                        gr.Markdown("""
+                        특수 능력을 선택하면 히어로의 DNA에 특별한 구조가 추가됩니다.
+                        - 자가 회복: 단백질 구조 복구 능력 강화
+                        - 원거리 공격: 특수한 구조적 돌출부 형성
+                        - 방어막 생성: 안정적인 보호층 구조 생성
+                        """)
                         special_ability = gr.CheckboxGroup(
                             choices=["자가 회복", "원거리 공격", "방어막 생성"],
                             label="특수 능력 선택"
                         )
                     
                     # 생성 버튼
-                    create_btn = gr.Button("히어로 생성!", variant="primary")
-
-                with gr.TabItem("Inputs"):
-                    gr.Markdown("""## INPUTS""")
-                    gr.Markdown("""#### Start Sequence
-                                Specify the protein length for complete unconditional generation, or scaffold a motif (or your name) using the custom sequence input""")
-                    seq_opt = gr.Radio(["protein length","custom sequence"], label="How would you like to specify the starting sequence?", value='protein length')
+                    create_btn = gr.Button("🧬 히어로 생성!", variant="primary", scale=2)
 
-                    sequence = gr.Textbox(label="custom sequence", lines=1, placeholder='AMINO ACIDS: A,C,D,E,F,G,H,I,K,L,M,N,P,Q,R,S,T,V,W,Y\n  MASK TOKEN: X', visible=False)
-                    seq_len = gr.Slider(minimum=5.0, maximum=250.0, label="protein length", value=100, visible=True)
+                with gr.TabItem("🧬 히어로 DNA 설계"):
+                    gr.Markdown("""
+                    ### 🧪 히어로 DNA 고급 설정
+                    히어로의 유전자 구조를 더 세밀하게 조정할 수 있습니다.
+                    """)
                     
-                    seq_opt.change(fn=toggle_seq_input,
-                                            inputs=[seq_opt],
-                                            outputs=[seq_len, sequence],
-                                            queue=False)
-
-                    gr.Markdown("""### Optional Parameters""")
-                    with gr.Accordion(label='Secondary Structure',open=True):
-                        gr.Markdown("""Try changing the sliders or inputing explicit secondary structure conditioning for each residue""")
-                        sec_str_opt = gr.Radio(["sliders","explicit"], label="How would you like to specify secondary structure?", value='sliders')
+                    seq_opt = gr.Radio(
+                        ["자동 설계", "직접 입력"],
+                        label="DNA 설계 방식",
+                        value="자동 설계"
+                    )
+
+                    sequence = gr.Textbox(
+                        label="DNA 시퀀스", 
+                        lines=1, 
+                        placeholder='사용 가능한 아미노산: A,C,D,E,F,G,H,I,K,L,M,N,P,Q,R,S,T,V,W,Y (X는 무작위)',
+                        visible=False
+                    )
+                    seq_len = gr.Slider(
+                        minimum=5.0, maximum=250.0, 
+                        label="DNA 길이", 
+                        value=100, 
+                        visible=True
+                    )
+                    
+                    with gr.Accordion(label='🦴 골격 구조 설정', open=True):
+                        gr.Markdown("""
+                        히어로의 기본 골격 구조를 설정합니다.
+                        - 나선형 구조: 유연하고 탄력있는 움직임
+                        - 병풍형 구조: 단단하고 강력한 힘
+                        - 고리형 구조: 빠르고 민첩한 움직임
+                        """)
+                        sec_str_opt = gr.Radio(
+                            ["슬라이더로 설정", "직접 입력"],
+                            label="골격 구조 설정 방식",
+                            value="슬라이더로 설정"
+                        )
 
-                        secondary_structure = gr.Textbox(label="secondary structure", lines=1, placeholder='HELIX = H  STRAND = S  LOOP = L  MASK = X(must be the same length as input sequence)', visible=False)
+                        secondary_structure = gr.Textbox(
+                            label="골격 구조",
+                            lines=1,
+                            placeholder='H:나선형, S:병풍형, L:고리형, X:자동설정',
+                            visible=False
+                        )
                         
                         with gr.Column():
-                            helix_bias = gr.Slider(minimum=0.0, maximum=0.05, label="helix bias", visible=True)
-                            strand_bias = gr.Slider(minimum=0.0, maximum=0.05, label="strand bias", visible=True)
-                            loop_bias = gr.Slider(minimum=0.0, maximum=0.20, label="loop bias", visible=True)
+                            helix_bias = gr.Slider(
+                                minimum=0.0, maximum=0.05,
+                                label="나선형 구조 비율",
+                                visible=True
+                            )
+                            strand_bias = gr.Slider(
+                                minimum=0.0, maximum=0.05,
+                                label="병풍형 구조 비율",
+                                visible=True
+                            )
+                            loop_bias = gr.Slider(
+                                minimum=0.0, maximum=0.20,
+                                label="고리형 구조 비율",
+                                visible=True
+                            )
+
+                with gr.TabItem("🧪 히어로 유전자 강화"):
+                    gr.Markdown("""
+                    ### ⚡ 기존 히어로의 DNA 활용
+                    강력한 히어로의 DNA 일부를 새로운 히어로에게 이식합니다.
+                    """)
                     
-                        sec_str_opt.change(fn=toggle_secondary_structure,
-                                                inputs=[sec_str_opt],
-                                                outputs=[helix_bias,strand_bias,loop_bias,secondary_structure],
-                                                queue=False)
-                        
-                    with gr.Accordion(label='Amino Acid Compositional Bias',open=False):
-                        gr.Markdown("""Bias sequence composition for particular amino acids by specifying the one letter code followed by the fraction to bias. This can be input as a list for example: W0.2,E0.1""")
-                        with gr.Row():
-                            aa_bias = gr.Textbox(label="aa bias", lines=1, placeholder='specify one letter AA and fraction to bias, for example W0.1 or M0.1,K0.1' )
-                            aa_bias_potential = gr.Textbox(label="aa bias scale", lines=1, placeholder='AA Bias potential scale (recomended range 1.0-5.0)')
-
-                    with gr.Accordion(label='Hydrophobic Bias',open=False):
-                        gr.Markdown("""Bias for or against hydrophobic composition, to get more soluble proteins, bias away with a negative target score (ex. -5)""")
-                        with gr.Row():
-                            hydrophobic_target_score = gr.Textbox(label="hydrophobic score", lines=1, placeholder='hydrophobic score to target (negative score is good for solublility)')
-                            hydrophobic_potential = gr.Textbox(label="hydrophobic potential scale", lines=1, placeholder='hydrophobic potential scale (recomended range 1.0-2.0)')
+                    pdb_id_code = gr.Textbox(
+                        label="히어로 DNA 코드",
+                        lines=1,
+                        placeholder='기존 히어로의 DNA 코드를 입력하세요 (예: 1DPX)',
+                        info="공개된 히어로 DNA 데이터베이스에서 코드를 찾을 수 있습니다"
+                    )
+                    
+                    contigs = gr.Textbox(
+                        label="이식할 DNA 영역",
+                        lines=1,
+                        placeholder='예시: 15,A3-10,20-30',
+                        info="이식하고 싶은 DNA 영역을 선택하고 새로운 DNA를 추가할 수 있습니다"
+                    )
                     
-                    with gr.Accordion(label='Diffusion Params',open=False):
-                        gr.Markdown("""Increasing T to more steps can be helpful for harder design challenges, sampling from different distributions can change the sequence and structural composition""")
-                        with gr.Row():
-                            num_steps = gr.Textbox(label="T", lines=1, placeholder='number of diffusion steps (25 or less will speed things up)')
-                            noise = gr.Dropdown(['normal','gmm2 [-1,1]','gmm3 [-1,0,1]'], label='noise type', value='normal')
-
-                with gr.TabItem("Motif Selection"):
-                    gr.Markdown("""### Motif Selection Preview""")
-                    gr.Markdown('Contigs explained: to grab residues (seq and str) on a pdb chain you will provide the chain letter followed by a range of residues as indexed in the pdb file for example (A3-10) is the syntax to select residues 3-10 on chain A (the chain always needs to be specified). To add diffused residues to either side of this motif you can specify a range or discrete value without a chain letter infront. To add 15 residues before the motif and 20-30 residues (randomly sampled) after use the following syntax: 15,A3-10,20-30 commas are used to separate regions selected from the pdb and designed (diffused) resiudes which will be added. ')
-                    pdb_id_code = gr.Textbox(label="PDB ID", lines=1, placeholder='INPUT PDB ID TO FETCH (ex. 1DPX)', visible=True)
-                    contigs = gr.Textbox(label="contigs", lines=1, placeholder='specify contigs to grab particular residues from pdb ()', visible=True)
-                    gr.Markdown('Using the same contig syntax, seq or str of input motif residues can be masked, allowing the model to hold strucutre fixed and design sequence or vice-versa')
                     with gr.Row():
-                        seq_mask = gr.Textbox(label='seq mask',lines=1,placeholder='input residues to mask sequence')
-                        str_mask = gr.Textbox(label='str mask',lines=1,placeholder='input residues to mask structure')
+                        seq_mask = gr.Textbox(
+                            label='능력 재설계',
+                            lines=1,
+                            placeholder='선택한 영역의 능력을 새롭게 디자인'
+                        )
+                        str_mask = gr.Textbox(
+                            label='구조 재설계',
+                            lines=1,
+                            placeholder='선택한 영역의 구조를 새롭게 디자인'
+                        )
+                    
                     preview_viewer = gr.HTML()
-                    rewrite_pdb = gr.File(label='PDB file')
-                    preview_btn = gr.Button("Preview Motif")
-
-                with gr.TabItem("MSA to PSSM"):
-                    gr.Markdown("""### MSA to PSSM Generation""")
-                    gr.Markdown('input either an MSA or PSSM to guide the model toward generating samples within your family of interest')
+                    rewrite_pdb = gr.File(label='히어로 DNA 파일')
+                    preview_btn = gr.Button("🔍 미리보기", variant="secondary")
+
+                with gr.TabItem("👑 히어로 가문"):
+                    gr.Markdown("""
+                    ### 🏰 위대한 히어로 가문의 유산
+                    강력한 히어로 가문의 특성을 계승하여 새로운 히어로를 만듭니다.
+                    """)
+                    
                     with gr.Row():
-                        fasta_msa = gr.File(label='MSA')
-                        input_pssm = gr.File(label='PSSM (.csv)')
-                    pssm = gr.File(label='Generated PSSM')
-                    pssm_view = gr.Plot(label='PSSM Viewer')
-                    pssm_gen_btn = gr.Button("Generate PSSM")
+                        fasta_msa = gr.File(
+                            label='가문 DNA 데이터',
+                            info="히어로 가문의 DNA 정보가 담긴 파일을 업로드하세요"
+                        )
+                        input_pssm = gr.File(
+                            label='가문 특성 데이터',
+                            info="이미 분석된 가문 특성 데이터를 가지고 있다면 업로드하세요"
+                        )
+                    
+                    pssm = gr.File(label='분석된 가문 특성')
+                    pssm_view = gr.Plot(label='가문 특성 분석 결과')
+                    pssm_gen_btn = gr.Button("✨ 가문 특성 분석", variant="secondary")
 
         with gr.Column():
             gr.Markdown("## 🦸‍♂️ 히어로 프로필")
@@ -651,18 +737,21 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             
             # 히어로 설명
             hero_description = gr.Textbox(label="히어로 특성", lines=3)
-            
 
-            gr.Markdown("""## OUTPUTS""")
-            gr.Markdown("""#### Confidence score for generated structure at each timestep""")
-            plddt_plot = gr.Plot(label='plddt at step t')
-            gr.Markdown("""#### Output protein sequnece""")
-            output_seq = gr.Textbox(label="sequence")
-            gr.Markdown("""#### Download PDB file""")
-            output_pdb = gr.File(label="PDB file")
-            gr.Markdown("""#### Structure viewer""")
+            gr.Markdown("## 🧬 히어로 DNA 분석 결과")
+            gr.Markdown("#### ⚡ DNA 안정성 점수")
+            plddt_plot = gr.Plot(label='안정성 분석')
+            gr.Markdown("#### 📝 DNA 시퀀스")
+            output_seq = gr.Textbox(label="DNA 서열")
+            gr.Markdown("#### 💾 DNA 데이터")
+            output_pdb = gr.File(label="DNA 파일")
+            gr.Markdown("#### 🔬 DNA 구조")
             output_viewer = gr.HTML()
 
+    # 이벤트 연결 부분은 동일하게 유지
+
+
+
     # 이벤트 연결
     preview_btn.click(get_motif_preview,[pdb_id_code, contigs],[preview_viewer, rewrite_pdb])
     pssm_gen_btn.click(get_pssm,[fasta_msa,input_pssm],[pssm_view, pssm])