diginoron commited on
Commit
26a2801
·
verified ·
1 Parent(s): a28416b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -123
app.py DELETED
@@ -1,123 +0,0 @@
1
- # app.py
2
- import os
3
- import pandas as pd
4
- import gradio as gr
5
- import comtradeapicall
6
- from huggingface_hub import InferenceClient
7
- from deep_translator import GoogleTranslator
8
- import spaces # برای مدیریت GPU کرایه‌ای
9
-
10
- # --- بارگذاری HS DATA از CSV گیت‌هاب ---
11
- HS_CSV_URL = (
12
- "https://raw.githubusercontent.com/"
13
- "datasets/harmonized-system/master/data/harmonized-system.csv"
14
- )
15
- hs_df = pd.read_csv(HS_CSV_URL, dtype=str)
16
-
17
- def get_product_name(hs_code: str) -> str:
18
- code4 = str(hs_code).zfill(4)
19
- row = hs_df[hs_df["hscode"] == code4]
20
- return row.iloc[0]["description"] if not row.empty else "–"
21
-
22
- # --- تابع دریافت واردات و پردازش ستون‌ها ---
23
- def get_importers(hs_code: str, year: str, month: str):
24
- product_name = get_product_name(hs_code)
25
- period = f"{year}{int(month):02d}"
26
- df = comtradeapicall.previewFinalData(
27
- typeCode='C', freqCode='M', clCode='HS', period=period,
28
- reporterCode=None, cmdCode=hs_code, flowCode='M',
29
- partnerCode=None, partner2Code=None,
30
- customsCode=None, motCode=None,
31
- maxRecords=500, includeDesc=True
32
- )
33
- if df is None or df.empty:
34
- return product_name, pd.DataFrame()
35
-
36
- # شناسایی ستون‌های مورد نیاز
37
- code_col = next((c for c in df.columns if 'code' in c.lower()), None)
38
- title_col = next((c for c in df.columns if 'title' in c.lower()), None)
39
- value_col = next((c for c in df.columns if 'value' in c.lower()), None)
40
- if not (code_col and title_col and value_col):
41
- return product_name, df
42
-
43
- df_sorted = df.sort_values(value_col, ascending=False).head(10)
44
- out = df_sorted[[code_col, title_col, value_col]]
45
- out.columns = ['کد کشور', 'نام کشور', 'ارزش CIF']
46
- return product_name, out
47
-
48
- # --- تابع تولید مشاوره تخصصی با GPU کرایه‌ای ---
49
- hf_token = os.getenv("HF_API_TOKEN")
50
- client = InferenceClient(token=hf_token)
51
- translator = GoogleTranslator(source='en', target='fa')
52
-
53
- @spaces.GPU
54
- def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
55
- if table_data is None or table_data.empty:
56
- return "ابتدا نمایش داده‌های واردات را انجام دهید."
57
-
58
- df_limited = table_data.head(10)
59
- table_str = df_limited.to_string(index=False)
60
- period = f"{year}/{int(month):02d}"
61
- prompt = (
62
- f"The following table shows the top {len(df_limited)} countries by CIF value importing HS code {hs_code} during {period}:\n"
63
- f"{table_str}\n\n"
64
- "Please provide a detailed and comprehensive analysis of market trends, risks, "
65
- "and opportunities for a new exporter entering this market."
66
- )
67
- try:
68
- outputs = client.text_generation(
69
- prompt=prompt,
70
- model="mistralai/Mixtral-8x7B-Instruct-v0.1",
71
- max_new_tokens=1024
72
- )
73
- return translator.translate(outputs)
74
- except Exception as e:
75
- return f"خطا در تولید مشاوره: {e}"
76
-
77
- # --- رابط کاربری Gradio با تنظیمات UI ---
78
- with gr.Blocks(css="""
79
- /* رنگ پس‌زمینه سفید و متن سیاه */
80
- body, .gradio-container { background-color: white !important; color: black !important; }
81
- /* پنهان کردن فوتر و لینک‌های Gradio */
82
- footer, .gradio-info { display: none !important; }
83
- """) as demo:
84
- # عنوان سفارشی راست‌چین
85
- gr.Markdown(
86
- "<div dir='rtl' style='text-align: right; font-family: IRANSans;'>"
87
- "<h2>هوش مصنوعی مشاوره صادراتی با HS Code محصول &ndash; ساخته شده توسط Diginoron</h2>"
88
- "</div>"
89
- )
90
-
91
- with gr.Row():
92
- inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
93
- inp_year = gr.Textbox(label="سال", placeholder="مثلاً 2023")
94
- inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
95
-
96
- btn_show = gr.Button("نمایش داده‌های واردات")
97
- out_name = gr.Markdown(label="<div dir='rtl' style='text-align: right; font-family: IRANSans;'>**نام محصول**</div>")
98
- out_table = gr.Dataframe(
99
- headers=['کد کشور', 'نام کشور', 'ارزش CIF'],
100
- datatype=["number", "text", "number"],
101
- interactive=True
102
- )
103
-
104
- btn_show.click(
105
- fn=get_importers,
106
- inputs=[inp_hs, inp_year, inp_month],
107
- outputs=[out_name, out_table]
108
- )
109
-
110
- btn_advice = gr.Button("ارائه مشاوره تخصصی")
111
- out_advice = gr.Textbox(label="<div dir='rtl' style='text-align: right; font-family: IRANSans;'>مشاوره تخصصی</div>", lines=8)
112
-
113
- btn_advice.click(
114
- fn=provide_advice,
115
- inputs=[out_table, inp_hs, inp_year, inp_month],
116
- outputs=out_advice
117
- )
118
-
119
- if __name__ == "__main__":
120
- demo.launch(
121
- share=True,
122
- show_api=False
123
- )