diginoron commited on
Commit
74b0cba
·
verified ·
1 Parent(s): 026d83f

Delete app (10).py

Browse files
Files changed (1) hide show
  1. app (10).py +0 -120
app (10).py DELETED
@@ -1,120 +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
- # ابتدا سعی در استفاده از ستون‌های استاندارد
38
- std_map = {
39
- 'کد کشور': 'ptCode',
40
- 'نام کشور': 'ptTitle',
41
- 'ارزش CIF': 'TradeValue'
42
- }
43
- code_col = std_map['کد کشور'] if 'ptCode' in df.columns else next((c for c in df.columns if 'code' in c.lower()), None)
44
- title_col= std_map['نام کشور'] if 'ptTitle' in df.columns else next((c for c in df.columns if 'title' in c.lower()), None)
45
- value_col= std_map['ارزش CIF'] if 'TradeValue' in df.columns else next((c for c in df.columns if 'value' in c.lower()), None)
46
-
47
- if not (code_col and title_col and value_col):
48
- # اگر نتوانست ستون‌ها را شناسایی کند، برگرداندن DataFrame خام
49
- return product_name, df
50
-
51
- # محدودسازی به 10 کشور برتر بر اساس ستون value_col
52
- df_sorted = df.sort_values(value_col, ascending=False).head(10)
53
-
54
- out = df_sorted[[code_col, title_col, value_col]]
55
- out.columns = ['کد کشور', 'نام کشور', 'ارزش CIF']
56
- return product_name, out
57
-
58
- # --- تابع تولید مشاوره تخصصی با GPU کرایه‌ای ---
59
- hf_token = os.getenv("HF_API_TOKEN")
60
- client = InferenceClient(token=hf_token)
61
- translator = GoogleTranslator(source='en', target='fa')
62
-
63
- @spaces.GPU
64
- def provide_advice(table_data: pd.DataFrame, hs_code: str, year: str, month: str):
65
- if table_data is None or table_data.empty:
66
- return "ابتدا نمایش داده‌های واردات را انجام دهید."
67
-
68
- # محدودسازی تعداد ردیف‌های ورودی به 10 (در صورت بیشتر)
69
- df_limited = table_data.head(10)
70
- table_str = df_limited.to_string(index=False)
71
- period = f"{year}/{int(month):02d}"
72
- prompt = (
73
- f"The following table shows the top {len(df_limited)} countries by CIF value importing HS code {hs_code} during {period}:\n"
74
- f"{table_str}\n\n"
75
- "Please provide a detailed and comprehensive analysis of market trends, risks, "
76
- "and opportunities for a new exporter entering this market."
77
- )
78
- try:
79
- outputs = client.text_generation(
80
- prompt=prompt,
81
- model="mistralai/Mixtral-8x7B-Instruct-v0.1",
82
- max_new_tokens=1024
83
- )
84
- return translator.translate(outputs)
85
- except Exception as e:
86
- return f"خطا در تولید مشاوره: {e}"
87
-
88
- # --- رابط کاربری Gradio ---
89
- with gr.Blocks() as demo:
90
- gr.Markdown("## تحلیل واردات بر اساس کد HS و ارائه مشاوره تخصصی")
91
-
92
- with gr.Row():
93
- inp_hs = gr.Textbox(label="کد HS", placeholder="مثلاً 1006")
94
- inp_year = gr.Textbox(label="سال", placeholder="مثلاً 2023")
95
- inp_month = gr.Textbox(label="ماه", placeholder="مثلاً 1 تا 12")
96
-
97
- btn_show = gr.Button("نمایش داده‌های واردات")
98
- out_name = gr.Markdown(label="**نام محصول**")
99
- out_table = gr.Dataframe(
100
- datatype="pandas",
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="مشاوره تخصصی", 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()