Update app.py

app.py

@@ -993,6 +993,63 @@ def refresh_data():
     else:
         return create_error_plot(), "<div>API 인증이 필요합니다.</div>", pd.DataFrame()
 
+
+
+def create_pie_chart(data, total_count, type_name="Spaces"):
+    korean_count = len(data)
+    other_count = total_count - korean_count
+    
+    fig = go.Figure(data=[go.Pie(
+        labels=[f'Korean {type_name}', f'Other {type_name}'],
+        values=[korean_count, other_count],
+        hole=.3,
+        marker_colors=['#FF6B6B', '#4ECDC4'],
+        textinfo='percent+value',
+        hovertemplate="<b>%{label}</b><br>" +
+                      "Count: %{value}<br>" +
+                      "Percentage: %{percent}<br>"
+    )])
+    
+    fig.update_layout(
+        title=f"Korean vs Other {type_name} Distribution",
+        showlegend=True,
+        height=400,
+        width=500
+    )
+    
+    return fig
+
+def create_registration_bar_chart(data, type_name="Spaces"):
+    # 등록자별 건수 집계
+    registrations = {}
+    for item in data:
+        creator = item.get('id', '').split('/')[0]  # ID의 첫 부분을 등록자로 간주
+        registrations[creator] = registrations.get(creator, 0) + 1
+    
+    # 정렬된 데이터 준비
+    sorted_data = sorted(registrations.items(), key=lambda x: x[1], reverse=True)
+    creators = [x[0] for x in sorted_data]
+    counts = [x[1] for x in sorted_data]
+    
+    fig = go.Figure(data=[go.Bar(
+        x=creators,
+        y=counts,
+        text=counts,
+        textposition='auto',
+        marker_color='#FF6B6B'
+    )])
+    
+    fig.update_layout(
+        title=f"Korean {type_name} Registrations by Creator",
+        xaxis_title="Creator ID",
+        yaxis_title="Number of Registrations",
+        showlegend=False,
+        height=400,
+        width=700
+    )
+    
+    return fig
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 🤗 허깅페이스 '한국(언어) 리더보드'
@@ -1004,35 +1061,70 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     with gr.Tab("Spaces Trending"):
         trending_plot = gr.Plot()
+        with gr.Row():
+            # 원형 그래프와 막대 그래프를 위한 컨테이너 추가
+            with gr.Column():
+                spaces_pie_chart = gr.Plot(label="Korean Spaces Distribution")
+            with gr.Column():
+                spaces_bar_chart = gr.Plot(label="Registrations by Creator")
         trending_info = gr.HTML()
         trending_df = gr.DataFrame()
     
     with gr.Tab("Models Trending"):
         models_plot = gr.Plot()
+        with gr.Row():
+            # 원형 그래프와 막대 그래프를 위한 컨테이너 추가
+            with gr.Column():
+                models_pie_chart = gr.Plot(label="Korean Models Distribution")
+            with gr.Column():
+                models_bar_chart = gr.Plot(label="Registrations by Creator")
         models_info = gr.HTML()
         models_df = gr.DataFrame()
     
     def refresh_all_data():
         spaces_results = get_spaces_data("trending")
         models_results = get_models_data()
-        return [*spaces_results, *models_results]
+        
+        # Spaces 차트 생성
+        spaces_pie = create_pie_chart(spaces_results[2], 500, "Spaces")  # DataFrame에서 데이터 추출
+        spaces_bar = create_registration_bar_chart(spaces_results[2], "Spaces")
+        
+        # Models 차트 생성
+        models_pie = create_pie_chart(models_results[2], 3000, "Models")  # DataFrame에서 데이터 추출
+        models_bar = create_registration_bar_chart(models_results[2], "Models")
+        
+        return [
+            spaces_results[0], spaces_results[1], spaces_results[2],
+            spaces_pie, spaces_bar,
+            models_results[0], models_results[1], models_results[2],
+            models_pie, models_bar
+        ]
     
     refresh_btn.click(
         refresh_all_data,
         outputs=[
             trending_plot, trending_info, trending_df,
-            models_plot, models_info, models_df
+            spaces_pie_chart, spaces_bar_chart,
+            models_plot, models_info, models_df,
+            models_pie_chart, models_bar_chart
         ]
     )
     
     # 초기 데이터 로드
-    spaces_results = get_spaces_data("trending")
-    models_results = get_models_data()
-    
-    trending_plot.value, trending_info.value, trending_df.value = spaces_results
-    models_plot.value, models_info.value, models_df.value = models_results
-    
+    initial_data = refresh_all_data()
     
+    # 초기값 설정
+    trending_plot.value = initial_data[0]
+    trending_info.value = initial_data[1]
+    trending_df.value = initial_data[2]
+    spaces_pie_chart.value = initial_data[3]
+    spaces_bar_chart.value = initial_data[4]
+    models_plot.value = initial_data[5]
+    models_info.value = initial_data[6]
+    models_df.value = initial_data[7]
+    models_pie_chart.value = initial_data[8]
+    models_bar_chart.value = initial_data[9]
+
 # Gradio 앱 실행
 demo.launch(
     server_name="0.0.0.0",