Update app.py

app.py

@@ -1,5 +1,5 @@
 import os,sys
-
+import traceback  # 상단에 추가
 # install required packages
 os.system('pip install plotly')  # plotly 설치
 os.system('pip install matplotlib')  # matplotlib 설치
@@ -909,6 +909,8 @@ def update_protein_display(chat_response):
             }
     return None
 
+
+
 def process_chat_and_generate(message, history):
     try:
         # 1. 프롬프트 분석 및 파라미터 설정
@@ -926,7 +928,7 @@ def process_chat_and_generate(message, history):
             secondary_structure=None,
             aa_bias=None,
             aa_bias_potential=None,
-            num_steps="25",  # 25단계 설정
+            num_steps="25",
             noise="normal",
             hydrophobic_target_score=str(params['hydrophobic_target_score']),
             hydrophobic_potential="2",
@@ -937,15 +939,21 @@ def process_chat_and_generate(message, history):
             rewrite_pdb=None
         )
         
-        
         # 3. 모든 단계의 결과 수집
         final_result = None
-        plddt_data = []  # plddt 데이터를 저장할 리스트 추가
+        plddt_data = []
+        
+        # 히어로 능력치 계산
+        stats = calculate_hero_stats(
+            helix_bias=params['helix_bias'],
+            strand_bias=params['strand_bias'],
+            loop_bias=params['loop_bias'],
+            hydrophobic_score=float(params['hydrophobic_target_score'])
+        )
         
         for result in generator:
             final_result = result
             if result[3]:  # plddt_plot이 존재하는 경우
-                # matplotlib 그래프에서 데이터 추출
                 ax = result[3].gca()
                 if ax.lines:
                     line = ax.lines[0]
@@ -957,12 +965,7 @@ def process_chat_and_generate(message, history):
                     {"role": "user", "content": message},
                     {"role": "assistant", "content": f"단백질 생성 중... {len(plddt_data)}단계 완료"}
                 ],
-                create_radar_chart(calculate_hero_stats(
-                    params['helix_bias'],
-                    params['strand_bias'],
-                    params['loop_bias'],
-                    params['hydrophobic_target_score']
-                )),
+                create_radar_chart(stats),
                 f"단백질 생성 진행 중... {len(plddt_data)}단계 완료",
                 result[0],  # output_seq
                 result[1],  # output_pdb
@@ -990,7 +993,7 @@ def process_chat_and_generate(message, history):
         - 소수성 점수: {params['hydrophobic_target_score']}
         
         생성이 완료되었습니다. 총 {len(plddt_data)}단계의 최적화를 수행했습니다.
-        최종 안정성 점수: {np.mean(plddt_data):.2f}
+        최종 안정성 점수: {np.mean(plddt_data) if plddt_data else 0:.2f}
         """
         
         # 6. 최종 결과 반환