Update app.py

app.py

@@ -58,9 +58,7 @@ def get_imports(hs_code: str, year: str, month: str):
             return product_name, df, "ستون‌های مورد نیاز در داده‌ها یافت نشد."
 
         df_sorted = df.sort_values(value_col, ascending=False).head(10)
-        out = df_sorted[[code_col, title_col, value_col]]
-        out.columns = ['کد کشور', 'نام کشور', 'ارزش CIF']
-        return product_name, out, ""
+        return product_name, df_sorted, ""  # بازگشت تمام ستون‌ها
     except Exception as e:
         return "–", pd.DataFrame(), f"خطا: {str(e)}"
 
@@ -128,6 +126,18 @@ button {
     border-radius: 5px;
     padding: 10px;
 }
+.gr-Dataframe table {
+    width: 100% !important;
+    table-layout: auto !important;
+}
+.gr-Dataframe th, .gr-Dataframe td {
+    min-width: 100px !important;  /* حداقل عرض برای ستون‌ها */
+    max-width: 250px !important;  /* حداکثر عرض برای جلوگیری از شلوغی */
+    white-space: normal !important;
+    word-wrap: break-word !important;
+    text-align: center !important;
+    padding: 5px !important;  /* فاصله داخلی برای خوانایی بهتر */
+}
 """
 
 # --- رابط کاربری سفارشی با Gradio Blocks ---
@@ -155,8 +165,7 @@ def create_custom_interface():
             out_name = gr.Markdown(label="**نام محصول**")
             out_table = gr.Dataframe(
                 datatype="pandas",
-                interactive=True,
-                headers=["کد کشور", "نام کشور", "ارزش CIF"]
+                interactive=True
             )
             out_message = gr.Markdown(label="پیام", visible=True)
 
@@ -181,4 +190,4 @@ def create_custom_interface():
 # --- راه‌اندازی رابط کاربری ---
 if __name__ == "__main__":
     demo = create_custom_interface()
-    demo.launch(show_api=False)  # حذف پارامتر theme
+    demo.launch(show_api=False)