Update app.py

app.py

@@ -845,6 +845,69 @@ def combined_generation(name, strength, flexibility, speed, defense, size, abili
         )
 
 
+def extract_parameters_from_chat(chat_response):
+    """챗봇 응답에서 파라미터 추출"""
+    try:
+        params = {
+            'sequence_length': 100,
+            'helix_bias': 0.02,
+            'strand_bias': 0.02,
+            'loop_bias': 0.1,
+            'hydrophobic_target_score': 0
+        }
+        
+        # 응답 텍스트에서 값 추출
+        if "길이:" in chat_response:
+            length_match = re.search(r'길이: (\d+)', chat_response)
+            if length_match:
+                params['sequence_length'] = int(length_match.group(1))
+                
+        if "알파 헬릭스 비율:" in chat_response:
+            helix_match = re.search(r'알파 헬릭스 비율: ([\d.]+)', chat_response)
+            if helix_match:
+                params['helix_bias'] = float(helix_match.group(1)) / 100
+                
+        if "베타 시트 비율:" in chat_response:
+            strand_match = re.search(r'베타 시트 비율: ([\d.]+)', chat_response)
+            if strand_match:
+                params['strand_bias'] = float(strand_match.group(1)) / 100
+                
+        if "루프 구조 비율:" in chat_response:
+            loop_match = re.search(r'루프 구조 비율: ([\d.]+)', chat_response)
+            if loop_match:
+                params['loop_bias'] = float(loop_match.group(1)) / 100
+                
+        if "소수성 점수:" in chat_response:
+            hydro_match = re.search(r'소수성 점수: ([-\d.]+)', chat_response)
+            if hydro_match:
+                params['hydrophobic_target_score'] = float(hydro_match.group(1))
+        
+        return params
+    except Exception as e:
+        print(f"파라미터 추출 중 오류: {str(e)}")
+        return None
+
+def update_protein_display(chat_response):
+    if "생성된 단백질 분석" in chat_response:
+        params = extract_parameters_from_chat(chat_response)
+        if params:
+            result = generate_protein(params)
+            stats = calculate_hero_stats(
+                helix_bias=params['helix_bias'],
+                strand_bias=params['strand_bias'],
+                loop_bias=params['loop_bias'],
+                hydrophobic_score=params['hydrophobic_target_score']
+            )
+            return {
+                hero_stats: create_radar_chart(stats),
+                hero_description: chat_response,
+                output_seq: result[0],
+                output_pdb: result[1],
+                output_viewer: display_pdb(result[1]),
+                plddt_plot: result[3]
+            }
+    return None
+
 def process_chat_and_generate(message, history):
     try:
         # 1. 프롬프트 분석 및 파라미터 설정
@@ -876,7 +939,15 @@ def process_chat_and_generate(message, history):
         # 3. 생성 결과 가져오기
         output_seq, output_pdb, structure_view, plddt_plot = next(generator)
         
-        # 4. 설명 생성
+        # 4. 히어로 능력치 계산
+        stats = calculate_hero_stats(
+            helix_bias=params['helix_bias'],
+            strand_bias=params['strand_bias'],
+            loop_bias=params['loop_bias'],
+            hydrophobic_score=params['hydrophobic_target_score']
+        )
+        
+        # 5. 설명 생성
         explanation = f"""
         요청하신 기능에 맞는 단백질을 생성했습니다:
         
@@ -891,16 +962,16 @@ def process_chat_and_generate(message, history):
         - 소수성 점수: {params['hydrophobic_target_score']}
         """
         
-        # 5. 챗봇 히스토리 업데이트
+        # 6. 챗봇 히스토리 업데이트
         new_history = history + [
             {"role": "user", "content": message},
             {"role": "assistant", "content": explanation}
         ]
         
-        # 6. 모든 결과 반환
+        # 7. 모든 결과 반환
         return (
             new_history,  # 챗봇 히스토리
-            create_radar_chart(calculate_hero_stats(params)),  # 능력치 차트
+            create_radar_chart(stats),  # 능력치 차트
             explanation,  # 설명
             output_seq,  # 단백질 서열
             output_pdb,  # PDB 파일
@@ -975,34 +1046,7 @@ def calculate_similarity(keywords, entry):
         print("Entry:", entry)
         return 0
 
-def extract_parameters_from_chat(chat_response):
-    """챗봇 응답에서 파라미터 추출"""
-    try:
-        params = {
-            'sequence_length': 100,
-            'helix_bias': 0.02,
-            'strand_bias': 0.02,
-            'loop_bias': 0.1,
-            'hydrophobic_target_score': 0
-        }
-        
-        # 응답 텍스트에서 값 추출
-        if "길이:" in chat_response:
-            length_match = re.search(r'길이: (\d+)', chat_response)
-            if length_match:
-                params['sequence_length'] = int(length_match.group(1))
-                
-        if "알파 헬릭스 비율:" in chat_response:
-            helix_match = re.search(r'알파 헬릭스 비율: ([\d.]+)', chat_response)
-            if helix_match:
-                params['helix_bias'] = float(helix_match.group(1)) / 100
-                
-        # 나머지 파라미터도 유사하게 추출
-        
-        return params
-    except Exception as e:
-        print(f"파라미터 추출 중 오류: {str(e)}")
-        return None
+
 
 def download_checkpoint_files():
     """필요한 체크포인트 파일 다운로드"""