Update app.py

app.py

@@ -909,10 +909,21 @@ def update_protein_display(chat_response):
             }
     return None
 
-
-
 def process_chat_and_generate(message, history):
     try:
+        # 초기 응답 생성
+        initial_response = "단백질 설계를 시작합니다. 잠시만 기다려주세요..."
+        
+        # 초기 응답 표시
+        yield (
+            history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": initial_response}
+            ],
+            None, None, None, None, None, None
+        )
+
+
         # 1. 프롬프트 분석 및 파라미터 설정
         analysis = analyze_prompt(message)
         similar_structures = search_protein_data(analysis, ds)
@@ -978,32 +989,43 @@ def process_chat_and_generate(message, history):
             
         output_seq, output_pdb, structure_view, plddt_plot = final_result
         
-        # 5. 최종 설명 생성
-        explanation = f"""
-        요청하신 기능에 맞는 단백질을 생성했습니다:
+
         
-        분석된 요구사항:
+
+        # 생성 완료 후 상세 설명 추가
+        final_explanation = f"""
+        단백질 설계가 완료되었습니다.
+
+        [분석 결과]
         {analysis}
-        
-        설계된 구조적 특징:
+
+        [구조적 특징]
         - 길이: {params['sequence_length']} 아미노산
         - 알파 헬릭스 비율: {params['helix_bias']*100:.1f}%
         - 베타 시트 비율: {params['strand_bias']*100:.1f}%
         - 루프 구조 비율: {params['loop_bias']*100:.1f}%
         - 소수성 점수: {params['hydrophobic_target_score']}
-        
-        생성이 완료되었습니다. 총 {len(plddt_data)}단계의 최적화를 수행했습니다.
-        최종 안정성 점수: {np.mean(plddt_data) if plddt_data else 0:.2f}
+
+        [생성 과정]
+        - 총 {len(plddt_data)}단계의 최적화 수행
+        - 최종 안정성 점수: {np.mean(plddt_data) if plddt_data else 0:.2f}
+        - 참조된 유사 구조: {len(similar_structures)}개
+
+        [기능적 특성]
+        - 주요 활성 부위: {analyze_active_sites(output_seq)}
+        - 예상되는 상호작용: {predict_interactions(params)}
+        - 안정성 평가: {evaluate_stability(plddt_data)}
+
+        3D 구조와 상세 분석 결과를 확인하실 수 있습니다.
         """
         
-        # 6. 최종 결과 반환
         return (
             history + [
                 {"role": "user", "content": message},
-                {"role": "assistant", "content": explanation}
+                {"role": "assistant", "content": final_explanation}
             ],
             create_radar_chart(stats),
-            explanation,
+            final_explanation,
             output_seq,
             output_pdb,
             structure_view,
@@ -1012,7 +1034,7 @@ def process_chat_and_generate(message, history):
         
     except Exception as e:
         print(f"Error in process_chat_and_generate: {str(e)}")
-        traceback.print_exc()  # 상세한 에러 정보 출력
+        traceback.print_exc()
         return (
             history + [
                 {"role": "user", "content": message},
@@ -1021,6 +1043,8 @@ def process_chat_and_generate(message, history):
             None, None, None, None, None, None
         )
 
+
+
 def extract_keywords(analysis):
     """분석 텍스트에서 키워드 추출"""
     try:
@@ -1145,8 +1169,8 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                 )
                 max_tokens = gr.Slider(
                     minimum=1,
-                    maximum=2048,
-                    value=512,
+                    maximum=3800,
+                    value=3800,
                     step=1,
                     label="최대 토큰 수"
                 )
@@ -1168,21 +1192,21 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
 
             # 탭 인터페이스
             with gr.Tabs():
-                with gr.TabItem("🦸‍♂️ 히어로 디자인"):
+                with gr.TabItem("🦸‍♂️ 커스텀 디자인"):
                     gr.Markdown("""
-                    ### ✨ 당신만의 특별한 히어로를 만들어보세요!
-                    각 능력치를 조절하면 히어로의 DNA가 자동으로 설계됩니다.
+                    ### ✨ 당신만의 특별한 커스텀을 만들어보세요!
+                    각 능력치를 조절하면 커스텀된 단백질이 자동으로 설계됩니다.
                     """)
                     
                     # 히어로 기본 정보
                     hero_name = gr.Textbox(
-                        label="히어로 이름", 
-                        placeholder="당신의 히어로 이름을 지어주세요!",
-                        info="히어로의 정체성을 나타내는 이름을 입력하세요"
+                        label="커스텀 이름", 
+                        placeholder="당신의 커스텀 단백질 이름을 지어��세요!",
+                        info="당신만의 정체성을 나타내는 이름을 입력하세요"
                     )
                     
                     # 능력치 설정
-                    gr.Markdown("### 💪 히어로 능력치 설정")
+                    gr.Markdown("### 💪 커스텀 능력치 설정")
                     with gr.Row():
                         strength = gr.Slider(
                             minimum=0.0, maximum=0.05, 
@@ -1214,15 +1238,15 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                     # 히어로 크기 설정
                     hero_size = gr.Slider(
                         minimum=50, maximum=200, 
-                        label="📏 히어로 크기", 
+                        label="📏 커스텀 단백질 크기", 
                         value=100,
-                        info="히어로의 전체적인 크기를 결정합니다"
+                        info="전체적인 크기를 결정합니다"
                     )
                     
                     # 특수 능력 설정
                     with gr.Accordion("🌟 특수 능력", open=False):
                         gr.Markdown("""
-                        특수 능력을 선택하면 히어로의 DNA에 특별한 구조가 추가됩니다.
+                        특수 능력을 선택하면 커스텀 단백질질에 특별한 구조가 추가됩니다.
                         - 자가 회복: 단백질 구조 복구 능력 강화
                         - 원거리 공격: 특수한 구조적 돌출부 형성
                         - 방어막 생성: 안정적인 보호층 구조 생성
@@ -1233,12 +1257,12 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                         )
                     
                     # 생성 버튼
-                    create_btn = gr.Button("🧬 히어로 생성!", variant="primary", scale=2)
+                    create_btn = gr.Button("🧬 커스텀 단백질 생성!", variant="primary", scale=2)
 
-                with gr.TabItem("🧬 히어로 DNA 설계"):
+                with gr.TabItem("🧬 커스텀 단백질 설계"):
                     gr.Markdown("""
-                    ### 🧪 히어로 DNA 고급 설정
-                    히어로의 유전자 구조를 더 세밀하게 조정할 수 있습니다.
+                    ### 🧪 커스텀 단백질 고급 설정
+                    유전자 구조를 더 세밀하게 조정할 수 있습니다.
                     """)
                     
                     seq_opt = gr.Radio(
@@ -1248,7 +1272,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                     )
 
                     sequence = gr.Textbox(
-                        label="DNA 시퀀스", 
+                        label="단백질 시퀀스", 
                         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
@@ -1262,7 +1286,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                     
                     with gr.Accordion(label='🦴 골격 구조 설정', open=True):
                         gr.Markdown("""
-                        히어로의 기본 골격 구조를 설정합니다.
+                        커스텀 단백질 기본 골격 구조를 설정합니다.
                         - 나선형 구조: 유연하고 탄력있는 움직임
                         - 병풍형 구조: 단단하고 강력한 힘
                         - 고리형 구조: 빠르고 민첩한 움직임
@@ -1297,9 +1321,9 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                                 visible=True
                             )
                     
-                    with gr.Accordion(label='🧬 DNA 구성 설정', open=False):
+                    with gr.Accordion(label='🧬 단백질 구성 설정', open=False):
                         gr.Markdown("""
-                        특정 아미노산의 비율을 조절하여 히어로의 특성을 강화할 수 있습니다.
+                        특정 아미노산의 비율을 조절하여 특성을 강화할 수 있습니다.
                         예시: W0.2,E0.1 (트립토판 20%, 글루탐산 10%)
                         """)
                         with gr.Row():
@@ -1316,7 +1340,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
 
                     with gr.Accordion(label='🌍 환경 적응력 설정', open=False):
                         gr.Markdown("""
-                        히어로의 환경 적응력을 조절합니다.
+                        환경 적응력을 조절합니다.
                         음수: 수중 활동에 특화, 양수: 지상 활동에 특화
                         """)
                         with gr.Row():
@@ -1347,24 +1371,24 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                                 value='normal'
                             )
                     
-                    design_btn = gr.Button("🧬 DNA 설계 생성!", variant="primary", scale=2)
+                    design_btn = gr.Button("🧬 단백질 설계 생성!", variant="primary", scale=2)
 
-                with gr.TabItem("🧪 히어로 유전자 강화"):
+                with gr.TabItem("🧪 커스텀 단백질 강화"):
                     gr.Markdown("""
-                    ### ⚡ 기존 히어로의 DNA 활용
-                    강력한 히어로의 DNA 일부를 새로운 히어로에게 이식합니다.
+                    ### ⚡ 기존 커스텀 단백질 활용
+                    기존 단백질 일부를 새로운 커스텀에게 이식합니다.
                     """)
                     
-                    gr.Markdown("공개된 히어로 DNA 데이터베이스에서 코드를 찾을 수 있습니다")
+                    gr.Markdown("공개된 커스텀 단백질 데이터베이스에서 코드를 찾을 수 있습니다")
                     pdb_id_code = gr.Textbox(
-                        label="히어로 DNA 코드",
+                        label="커스텀 단백질 코드",
                         lines=1,
-                        placeholder='기존 히어로의 DNA 코드를 입력하세요 (예: 1DPX)'
+                        placeholder='기존 커스텀 단백질 코드를 입력하세요 (예: 1DPX)'
                     )
                     
-                    gr.Markdown("이식하고 싶은 DNA 영역을 선택하고 새로운 DNA를 추가할 수 있습니다")
+                    gr.Markdown("이식하고 싶은 단백질 영역을 선택하고 새로운 단백질을 추가할 수 있습니다")
                     contigs = gr.Textbox(
-                        label="이식할 DNA 영역",
+                        label="이식할 단백질 영역",
                         lines=1,
                         placeholder='예시: 15,A3-10,20-30'
                     )
@@ -1382,19 +1406,19 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
                         )
                     
                     preview_viewer = gr.HTML()
-                    rewrite_pdb = gr.File(label='히어로 DNA 파일')
+                    rewrite_pdb = gr.File(label='커스텀 단백질 파일')
                     preview_btn = gr.Button("🔍 미리보기", variant="secondary")
-                    enhance_btn = gr.Button("⚡ 강화된 히어로 생성!", variant="primary", scale=2)
+                    enhance_btn = gr.Button("⚡ 강화된 커스텀 단백질 생성!", variant="primary", scale=2)
 
-                with gr.TabItem("👑 히어로 가문"):
+                with gr.TabItem("👑 커스텀 단백질 족보"):
                     gr.Markdown("""
-                    ### 🏰 위대한 히어로 가문의 유산
-                    강력한 히어로 가문의 특성을 계승하여 새로운 히어로를 만듭니다.
+                    ### 🏰 위대한 커스텀 단백질 가문의 유산
+                    강력한 특성을 계승하여 새로운 커스텀 단백질을 만듭니다.
                     """)
                     
                     with gr.Row():
                         with gr.Column():
-                            gr.Markdown("히어로 가문의 DNA 정보가 담긴 파일을 업로드하세요")
+                            gr.Markdown("커스텀 단백질 정보가 담긴 파일을 업로드하세요")
                             fasta_msa = gr.File(label='가문 DNA 데이터')
                         with gr.Column():
                             gr.Markdown("이미 분석된 가문 특성 데이터가 있다면 업로드하세요")
@@ -1407,18 +1431,18 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
 
         # 오른쪽 열: 결과 표시
         with gr.Column(scale=1):
-            gr.Markdown("## 🦸‍♂️ 히어로 프로필")
+            gr.Markdown("## 🦸‍♂️ 커스텀 단백질 프로필")
             hero_stats = gr.Plot(label="능력치 분석")
-            hero_description = gr.Textbox(label="히어로 특성", lines=3)
+            hero_description = gr.Textbox(label="커스텀 단백질 특성", lines=3)
             
-            gr.Markdown("## 🧬 히어로 DNA 분석 결과")
-            gr.Markdown("#### ⚡ DNA 안정성 점수")
+            gr.Markdown("## 🧬 커스텀 단백질 분석 결과")
+            gr.Markdown("#### ⚡ 커스텀 단백질 안정성 점수")
             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 구조")
+            gr.Markdown("#### 📝 커스텀 단백질 시퀀스")
+            output_seq = gr.Textbox(label="커스텀 단백질 서열")
+            gr.Markdown("#### 💾 커스텀 단백질 데이터")
+            output_pdb = gr.File(label="커스텀 단백질 파일")
+            gr.Markdown("#### 🔬 커스텀 단백질 구조")
             output_viewer = gr.HTML()
 
     # 이벤트 연결