Update app.py

app.py

@@ -12,10 +12,11 @@ from datetime import datetime, timedelta
 # 한국어 처리를 위한 KoSentence-BERT 모델 로드
 model = SentenceTransformer('jhgan/ko-sbert-sts')
 
-# 나눔바른고딕 폰트 설정
 font_path = "./NanumBarunGothic.ttf"
 font_prop = fm.FontProperties(fname=font_path)
-plt.rcParams['font.family'] = font_prop.get_name()
+plt.rcParams['font.family'] = 'NanumBarunGothic'
+plt.rcParams['font.sans-serif'] = ['NanumBarunGothic']
+fm.fontManager.addfont(font_path)
 
 # 전역 변수
 global_recommendations = None
@@ -31,16 +32,15 @@ def create_csv_string(recommendations):
         writer.writerow(rec)
     return output.getvalue()
 
-# 차트 생성 함수
 def create_chart(G):
     plt.figure(figsize=(10, 8))
     pos = nx.spring_layout(G)
     nx.draw(G, pos, with_labels=True, node_color='lightblue', node_size=3000, font_size=10, font_weight='bold', edge_color='gray')
     plt.title("직원과 프로그램 간의 관계", fontsize=14, fontweight='bold')
-    plt.tight_layout()
+    plt.tight_layout(pad=1.0)
     
     buf = io.BytesIO()
-    plt.savefig(buf, format='png')
+    plt.savefig(buf, format='png', bbox_inches='tight')
     buf.seek(0)
     plt.close()
     return buf
@@ -76,8 +76,11 @@ def validate_and_get_columns(employee_df, program_df):
     return None, employee_cols, program_cols
 
 # 유튜브 데이터 열 선택 함수
+
 def select_youtube_columns(youtube_file):
     global youtube_columns
+    if youtube_file is None:
+        return [gr.Dropdown(choices=[], value="") for _ in range(4)]
     youtube_df = pd.read_csv(youtube_file.name)
     required_youtube_cols = ["title", "description", "url", "upload_date"]
     youtube_columns = auto_match_columns(youtube_df, required_youtube_cols)
@@ -194,9 +197,9 @@ def hybrid_rag(employee_file, program_file, youtube_file, title_col, description
     
     # 결과 테이블 데이터프레임 생성
     result_df = pd.DataFrame(recommendation_rows, columns=["Employee ID", "Employee Name", "Recommended Programs", "Recommended YouTube Content"])
-    
-    return result_df, chart_buffer, gr.File(value=global_csv_string, visible=True), gr.Button.update(visible=True)
 
+    return result_df, chart_buffer, gr.File(value=global_csv_string, visible=True), gr.Button(value="CSV 다운로드", visible=True)
+    
 # 채팅 응답 함수
 def chat_response(message, history):
     global global_recommendations