Update app.py

app.py

@@ -5,8 +5,7 @@ import gradio as gr
 import comtradeapicall
 from openai import OpenAI
 from deep_translator import GoogleTranslator
-import matplotlib.pyplot as plt
-import spaces
+import spaces  # برای مدیریت GPU کرایه‌ای
 
 # --- بارگذاری داده‌های HS Code ---
 HS_CSV_URL = (
@@ -50,22 +49,8 @@ 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(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
-
-# --- مشاوره با GPT + ترجمه فارسی ---
-openai_client = OpenAI(api_key=os.getenv("OPENAI"))
+# --- اتصال به OpenAI و مترجم ---
+openai_client = OpenAI(api_key=os.getenv("OPENAI"))  # سکرت از محیط
 translator = GoogleTranslator(source='en', target='fa')
 
 @spaces.GPU
@@ -108,24 +93,15 @@ 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="نمودار ۵ کشور اول واردکننده")
-
-    # دکمه اول: دو خروجی با هم
-    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=combined_output,
+        fn=get_importers,
         inputs=[inp_hs, inp_year, inp_month],
-        outputs=[out_name, out_table, out_chart]
+        outputs=[out_name, out_table]
     )
 
-    # دکمه دوم: مشاوره تخصصی
     btn_advice = gr.Button("ارائه مشاوره تخصصی")
     out_advice = gr.Textbox(label="مشاوره تخصصی", lines=8)