Update app.py

app.py

@@ -5,7 +5,8 @@ import gradio as gr
 import comtradeapicall
 from openai import OpenAI
 from deep_translator import GoogleTranslator
-import spaces  # برای مدیریت GPU کرایه‌ای
+import matplotlib.pyplot as plt
+import spaces  # برای GPU کرایه‌ای
 
 # --- بارگذاری داده‌های HS Code ---
 HS_CSV_URL = (
@@ -49,8 +50,22 @@ def get_importers(hs_code: str, year: str, month: str):
     out.columns = ['کد کشور', 'نام کشور', 'ارزش CIF']
     return product_name, out
 
+# --- ترسیم نمودار ۵ کشور اول ---
+def plot_top5_chart(table_data: pd.DataFrame):
+    if table_data is None or table_data.empty:
+        return None
+
+    df_sorted = table_data.sort_values("ارزش CIF", ascending=False).head(5)
+
+    fig, ax = plt.subplots()
+    ax.bar(df_sorted["نام کشور"], df_sorted["ارزش CIF"], color="skyblue")
+    ax.set_ylabel("ارزش CIF (USD)")
+    ax.set_title("۵ کشور اول واردکننده")
+    ax.tick_params(axis='x', rotation=45)
+    return fig
+
 # --- اتصال به OpenAI و مترجم ---
-openai_client = OpenAI(api_key=os.getenv("OPENAI"))  # سکرت از محیط
+openai_client = OpenAI(api_key=os.getenv("OPENAI"))
 translator = GoogleTranslator(source='en', target='fa')
 
 @spaces.GPU
@@ -93,8 +108,10 @@ with gr.Blocks() as demo:
         inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
 
     btn_show = gr.Button("نمایش داده‌های واردات")
+
     out_name = gr.Markdown(label="**نام محصول**")
     out_table = gr.Dataframe(datatype="pandas", interactive=True)
+    out_chart = gr.Plot(label="نمودار ۵ کشور اول واردکننده")
 
     btn_show.click(
         fn=get_importers,
@@ -102,6 +119,12 @@ with gr.Blocks() as demo:
         outputs=[out_name, out_table]
     )
 
+    btn_show.click(
+        fn=plot_top5_chart,
+        inputs=[out_table],
+        outputs=[out_chart]
+    )
+
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
     out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)