Update app.py

app.py

@@ -6,7 +6,7 @@ import comtradeapicall
 from openai import OpenAI
 from deep_translator import GoogleTranslator
 import matplotlib.pyplot as plt
-import spaces  # برای GPU کرایه‌ای
+import spaces
 
 # --- بارگذاری داده‌های HS Code ---
 HS_CSV_URL = (
@@ -50,21 +50,21 @@ 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()
+    fig, ax = plt.subplots(figsize=(6, 4))
     ax.bar(df_sorted["نام کشور"], df_sorted["ارزش CIF"], color="skyblue")
     ax.set_ylabel("ارزش CIF (USD)")
     ax.set_title("۵ کشور اول واردکننده")
     ax.tick_params(axis='x', rotation=45)
+    plt.tight_layout()
     return fig
 
-# --- اتصال به OpenAI و مترجم ---
+# --- مشاوره با GPT + ترجمه فارسی ---
 openai_client = OpenAI(api_key=os.getenv("OPENAI"))
 translator = GoogleTranslator(source='en', target='fa')
 
@@ -113,18 +113,19 @@ with gr.Blocks() as demo:
     out_table = gr.Dataframe(datatype="pandas", interactive=True)
     out_chart = gr.Plot(label="نمودار ۵ کشور اول واردکننده")
 
-    btn_show.click(
-        fn=get_importers,
-        inputs=[inp_hs, inp_year, inp_month],
-        outputs=[out_name, out_table]
-    )
+    # دکمه اول: دو خروجی با هم
+    def combined_output(hs_code, year, month):
+        name, table = get_importers(hs_code, year, month)
+        chart = plot_top5_chart(table)
+        return name, table, chart
 
     btn_show.click(
-        fn=plot_top5_chart,
-        inputs=[out_table],
-        outputs=[out_chart]
+        fn=combined_output,
+        inputs=[inp_hs, inp_year, inp_month],
+        outputs=[out_name, out_table, out_chart]
     )
 
+    # دکمه دوم: مشاوره تخصصی
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
     out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)