Update app.py

app.py

@@ -53,27 +53,21 @@ translator = GoogleTranslator(source='en', target='fa')
 @spaces.GPU
 def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
     if table_data is None or table_data.empty:
-        return "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">ابتدا نمایش داده‌های واردات را انجام دهید.</div>"
+        return "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">ابتدا نمایش داده‌های واردات را انجام دهید.</div>"
 
     df_limited = table_data.head(10)
     table_str = df_limited.to_string(index=False)
     period = f"{year}/{int(month):02d}"
     prompt = f"""
 لطفاً یک گزارش تحلیلی به زبان فارسی و با چینش راست‌به‌چپ تولید کن، شامل دو بخش:
-1. جدول خلاصه (Markdown) 10 کشور برتر با ستون‌های رتبه، نام کشور، ارزش CIF و سه تأمین‌کننده برتر.
-2. نکات کلیدی و پیشنهادات به‌صورت بولت‌پوینت با تراز راست.
 
-- متن خروجی را درون یک <div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\"> قرار بده.
+1. **جدول خلاصه** (Markdown) ۱۰ کشور برتر با ستون‌های «رتبه»، «نام کشور»، «ارزش CIF» و «سه تأمین‌کننده برتر».\n
+2. **نکات کلیدی و پیشنهادات** به‌صورت بولت‌پوینت با تراز راست.
 
-داده‌ها:
+
+**داده‌ها:**
 {table_str}
-""" 10 کشور برتر با ستون‌های رتبه، نام کشور، ارزش CIF و سه تأمین‌کننده برتر."
-        "\n2. نکات کلیدی و پیشنهادات به‌صورت بولت‌پوینت با تراز راست."
-        "\n
-- متن خروجی را درون یک <div dir=\\\"rtl\\\" style=\\\"text-align: right; font-family: IRANSans;\\\"> قرار بده."
-        f"\n
-داده‌ها:\n{table_str}"
-    )
+"""
     try:
         outputs = client.text_generation(
             prompt=prompt,
@@ -81,14 +75,19 @@ def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str
             max_new_tokens=2048
         )
         persian = translator.translate(outputs)
-        # قرار دادن در کانتینر HTML راست‌چین
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">{persian}</div>"
+        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">{persian}</div>"
     except Exception as e:
-        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">خطا در تولید مشاوره: {e}</div>"
+        return f"<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">خطا در تولید مشاوره: {e}</div>"
 
 # --- رابط کاربری Gradio ---
-with gr.Blocks() as demo:
-    gr.Markdown("<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\"><h2>تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی</h2></div>", elem_id="title")
+with gr.Blocks(css="""
+body, .gradio-container {background-color: white; color: black;}
+""") as demo:
+    gr.Markdown(
+        "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">"
+        "<h2>هوش مصنوعی مشاوره صادراتی با HS Code محصول – ساخته شده توسط Diginoron</h2>"
+        "</div>"
+    )
 
     with gr.Row():
         inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
@@ -96,9 +95,14 @@ with gr.Blocks() as demo:
         inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
 
     btn_show = gr.Button("نمایش داده‌های واردات")
-    out_name = gr.Markdown(label="**<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans;\">نام محصول</div>")
+    out_name = gr.Markdown(
+        "<div dir=\"rtl\" style=\"text-align: right; font-family: IRANSans; color: black; background: white;\">"
+        "**نام محصول**"
+        "</div>"
+    )
     out_table = gr.Dataframe(
-        datatype="pandas",
+        headers=["کد کشور", "نام کشور", "ارزش CIF"],
+        datatype=["text", "text", "text"],
         interactive=True
     )
 
@@ -118,4 +122,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()