Update app.py

app.py

@@ -469,22 +469,18 @@ def generate_hero(name, strength, flexibility, speed, defense, size, abilities):
             
         if final_result is None:
             raise Exception("생성 결과가 없습니다")
-        
-        output_seq, output_pdb, structure_view, plddt_plot = final_result
             
         # 능력치 계산
         stats = calculate_hero_stats(flexibility, strength, speed, defense)
         
-        # 결과 반환
+        # 결과 반환 (3D 구조 제외)
         return (
-            structure_view,  # 3D 구조
             create_radar_chart(stats),  # 능력치 차트
             generate_hero_description(name, stats, abilities)  # 히어로 설명
         )
     except Exception as e:
-        print(f"Error in generate_hero: {str(e)}")  # 디버깅을 위한 출력 추가
+        print(f"Error in generate_hero: {str(e)}")
         return (
-            gr.HTML("에러가 발생했습니다"),
             None,
             f"에러: {str(e)}"
         )
@@ -634,17 +630,16 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
 
         with gr.Column():
             gr.Markdown("## 🦸‍♂️ 히어로 프로필")
-            
-            # 히어로 시각화
-            hero_structure = gr.HTML(label="히어로 구조")
-            
-            # 능력치 레이더 차트
+    
+    # 히어로 시각화 부분 제거 (hero_structure 삭제)
+    
+    # 능력치 레이더 차트
             hero_stats = gr.Plot(label="능력치 분석")
-            
-            # 히어로 설명
+    
+    # 히어로 설명
             hero_description = gr.Textbox(label="히어로 특성", lines=3)
-            
-            # 다운로드 버튼
+    
+    # 다운로드 버튼
             download_btn = gr.Button("히어로 데이터 다운로드")
 
             gr.Markdown("""## OUTPUTS""")
@@ -657,15 +652,20 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             gr.Markdown("""#### Structure viewer""")
             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])
+
+
     create_btn.click(
         generate_hero,
         inputs=[hero_name, strength, flexibility, speed, defense, hero_size, special_ability],
-        outputs=[hero_structure, hero_stats, hero_description]
+        outputs=[hero_stats, hero_description]  # hero_structure 제거
     )
 
+    
     btn = gr.Button("GENERATE")
     btn.click(protein_diffusion_model, 
                 [sequence,