Update app.py

app.py

@@ -38,25 +38,6 @@ def generate_colors(n):
     HSV_tuples = [(x * 1.0 / n, 0.5, 0.9) for x in range(n)]
     return ['#%02x%02x%02x' % tuple(int(x*255) for x in colorsys.hsv_to_rgb(*hsv)) for hsv in HSV_tuples]
 
-def create_network_graph(topic_results, num_words=10):
-    G = nx.Graph()
-    colors = generate_colors(len(topic_results))
-    
-    for idx, topic in enumerate(topic_results):
-        words = topic['lda_words'][:num_words]
-        color = colors[idx]
-        
-        for word in words:
-            if not G.has_node(word):
-                G.add_node(word, color=color)
-            
-        for i in range(len(words)):
-            for j in range(i+1, len(words)):
-                if not G.has_edge(words[i], words[j]):
-                    G.add_edge(words[i], words[j])
-    
-    return G
-
 def plot_network_graph(G):
     # 한국어 폰트 설정
     font_path = "./NanumBarunGothic.ttf"
@@ -64,16 +45,19 @@ def plot_network_graph(G):
     plt.rc('font', family=font_prop.get_name())
 
     plt.figure(figsize=(12, 8))
-    pos = nx.spring_layout(G, k=0.5, iterations=50)
+    
+    # 고정된 레이아웃 사용
+    pos = nx.spring_layout(G, k=0.5, iterations=50, seed=42)
     
     node_colors = [G.nodes[node]['color'] for node in G.nodes()]
     
-    nx.draw(G, pos, node_color=node_colors, with_labels=True, node_size=1000, font_size=8,
-            font_weight='bold', edge_color='gray', width=0.5, font_family=font_prop.get_name())
+    nx.draw(G, pos, node_color=node_colors, with_labels=False, node_size=1000, 
+            edge_color='gray', width=0.5)
     
-    for node, (x, y) in pos.items():
-        plt.text(x, y, node, fontsize=8, fontweight='bold', ha='center', va='center',
-                 fontproperties=font_prop)
+    # 라벨 위치 조정
+    label_pos = {k: (v[0], v[1] + 0.02) for k, v in pos.items()}
+    nx.draw_networkx_labels(G, label_pos, font_size=8, font_weight='bold', 
+                            font_family=font_prop.get_name())
     
     plt.title("Network graph", fontsize=16)
     plt.axis('off')
@@ -85,6 +69,25 @@ def plot_network_graph(G):
     
     return img_bytes
 
+def create_network_graph(topic_results, num_words=10):
+    G = nx.Graph()
+    colors = generate_colors(len(topic_results))
+    
+    for idx, topic in enumerate(topic_results):
+        words = topic['lda_words'][:num_words]
+        color = colors[idx]
+        
+        for word in words:
+            if not G.has_node(word):
+                G.add_node(word, color=color)
+            
+        for i in range(len(words)):
+            for j in range(i+1, len(words)):
+                if not G.has_edge(words[i], words[j]):
+                    G.add_edge(words[i], words[j])
+    
+    return G
+
 # 헤더 스타일 변경
 st.markdown("""
 <style>