leilaghomashchi commited on
Commit
ecb6c42
·
verified ·
1 Parent(s): d645475

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -398
app.py DELETED
@@ -1,398 +0,0 @@
1
- import gradio as gr
2
- import re
3
- import os
4
- import requests
5
- import logging
6
- from typing import Dict, List, Tuple
7
-
8
- logging.basicConfig(level=logging.INFO)
9
- logger = logging.getLogger(__name__)
10
-
11
- class AnonymizerAdvanced:
12
- """ناشناس‌ساز پیشرفته با روش‌های متعدد"""
13
-
14
- def __init__(self, cerebras_key: str = None, gpt_key: str = None):
15
- self.cerebras_key = cerebras_key or os.getenv("CEREBRAS_API_KEY")
16
- self.gpt_key = gpt_key or os.getenv("OPENAI_API_KEY")
17
- self.mapping_table = {} # {placeholder: original_text}
18
- self.reverse_mapping = {} # {original_text: placeholder}
19
-
20
- logger.info("✅ Anonymizer Advanced مقداردهی شد")
21
-
22
-
23
- def anonymize_with_cerebras(self, text: str) -> Tuple[str, Dict]:
24
- """ناشناس‌سازی با Cerebras"""
25
- logger.info("🧠 روش Cerebras...")
26
-
27
- if not self.cerebras_key:
28
- logger.error("❌ Cerebras API Key موجود نیست")
29
- raise ValueError("Cerebras API Key مورد نیاز است")
30
-
31
- try:
32
- prompt = f"""متن زیر را ناشناس کنید. قوانین:
33
- 1. اسامی اشخاص → person-01, person-02, ...
34
- 2. نام شرکت‌ها/سازمان‌ها → company-01, company-02, ...
35
- 3. مقادیر پولی → amount-01, amount-02, ...
36
- 4. درصدها → percent-01, percent-02, ...
37
- 5. فقط این توکن‌ها استفاده کنید
38
- 6. شماره‌های نسخه را درست حفظ کنید
39
- 7. اگر موجودیت تکرار شود از شماره قدیمی استفاده کنید
40
-
41
- متن:
42
- {text}
43
-
44
- خروجی: فقط متن ناشناس شده"""
45
-
46
- response = requests.post(
47
- "https://api.cerebras.ai/v1/chat/completions",
48
- headers={
49
- "Authorization": f"Bearer {self.cerebras_key}",
50
- "Content-Type": "application/json"
51
- },
52
- json={
53
- "model": "llama-3.3-70b",
54
- "messages": [{"role": "user", "content": prompt}],
55
- "max_tokens": 4096,
56
- "temperature": 0.1
57
- },
58
- timeout=60
59
- )
60
-
61
- if response.status_code == 200:
62
- anonymized_text = response.json()['choices'][0]['message']['content'].strip()
63
- logger.info("✅ Cerebras: موفق")
64
-
65
- # استخراج mapping از متن
66
- self._extract_mapping_from_text(text, anonymized_text)
67
- return anonymized_text, self.mapping_table
68
- else:
69
- logger.error(f"❌ Cerebras Error: {response.status_code}")
70
- raise Exception(f"Cerebras API Error: {response.status_code}")
71
-
72
- except Exception as e:
73
- logger.error(f"❌ Cerebras Exception: {e}")
74
- raise
75
-
76
-
77
- def _extract_mapping_from_text(self, original: str, anonymized: str):
78
- """استخراج mapping از متن‌های اصلی و ناشناس شده"""
79
- # الگوهای موجودیت
80
- patterns = {
81
- 'person': r'\b[ء-ي]+\s+[ء-ي]+(?:\s+[ء-ي]+)*\b',
82
- 'company': r'(?:شرکت|بانک|سازمان|گروه|هلدینگ)\s+[ء-ي]+(?:\s+[ء-ي]+)*',
83
- 'amount': r'\d+\s*(?:میلیارد|میلیون|هزار|تومان|ریال|دلار|تن)',
84
- 'percent': r'\d+\s*(?:درصد|%|درصدی)',
85
- }
86
-
87
- for entity_type, pattern in patterns.items():
88
- for match in re.finditer(pattern, original):
89
- text_match = match.group()
90
- tokens = re.findall(f'{entity_type}-\\d+', anonymized)
91
- if tokens:
92
- for token in tokens:
93
- if token not in self.mapping_table:
94
- self.mapping_table[token] = text_match
95
- self.reverse_mapping[text_match] = token
96
-
97
- def analyze_with_gpt(self, anonymized_text: str, analysis_prompt: str = None) -> str:
98
- """اجرای پرامپت‌های درون متن ناشناس‌سازی شده با ChatGPT"""
99
- logger.info("🤖 ChatGPT اجرای پرامپت...")
100
-
101
- if not self.gpt_key:
102
- logger.warning("⚠️ GPT API Key نیست")
103
- return "❌ API Key موجود نیست"
104
-
105
- try:
106
- # اگر پرامپت تحلیلی نداده شد، پرامپت پیش‌فرض استفاده شود
107
- if not analysis_prompt:
108
- analysis_prompt = """لطفاً این متن را تحلیل کنید:
109
- 1. خلاصه اطلاعات مالی
110
- 2. کیان‌های اصلی (با توکن‌های ناشناس شده)
111
- 3. رابطه‌های مهم
112
- 4. داده‌های عددی و درصدها"""
113
-
114
- # متن ارسالی شامل متن ناشناس‌ساز�� شده + دستورات
115
- user_message = f"""متن ناشناس‌سازی شده:
116
- ---
117
- {anonymized_text}
118
- ---
119
-
120
- دستورات:
121
- {analysis_prompt}
122
-
123
- توکن‌های ناشناس را حتماً حفظ کن. فقط متن تحلیل‌شده را برگردان."""
124
-
125
- logger.info(f"📋 متن ارسالی به ChatGPT:\n{user_message}\n")
126
-
127
- response = requests.post(
128
- "https://api.openai.com/v1/chat/completions",
129
- headers={
130
- "Authorization": f"Bearer {self.gpt_key}",
131
- "Content-Type": "application/json"
132
- },
133
- json={
134
- "model": "gpt-4o-mini",
135
- "messages": [
136
- {
137
- "role": "system",
138
- "content": """شما دستیار تحلیل متون مالی‌اند که:
139
- 1. متن‌های ناشناس‌سازی شده را تحلیل می‌کنید
140
- 2. توکن‌های ناشناس (person-01, company-01, amount-01, percent-01) را حفظ می‌کنید
141
- 3. دستورات کاربر را اجرا می‌کنید
142
- 4. فقط نتیجه تحلیل را برگردانید بدون توضیح اضافی"""
143
- },
144
- {
145
- "role": "user",
146
- "content": user_message
147
- }
148
- ],
149
- "max_tokens": 4096,
150
- "temperature": 0.3
151
- },
152
- timeout=60
153
- )
154
-
155
- if response.status_code == 200:
156
- gpt_response = response.json()['choices'][0]['message']['content']
157
- logger.info("✅ ChatGPT: دستورات اجرا شدند")
158
- logger.info(f"📤 پاسخ ChatGPT:\n{gpt_response}\n")
159
- return gpt_response
160
- else:
161
- logger.error(f"❌ GPT Error: {response.status_code}")
162
- return "❌ خطای ChatGPT"
163
-
164
- except Exception as e:
165
- logger.error(f"❌ GPT Exception: {e}")
166
- return f"❌ خطا: {str(e)}"
167
-
168
- def restore_text(self, anonymized_text: str) -> str:
169
- """بازگردانی متن اصلی"""
170
- logger.info("🔄 بازگردانی...")
171
-
172
- restored = anonymized_text
173
- for placeholder, original in sorted(self.mapping_table.items()):
174
- restored = restored.replace(placeholder, original)
175
-
176
- logger.info("✅ بازگردانی کامل")
177
- return restored
178
-
179
- def get_mapping_table_md(self) -> str:
180
- """تبدیل جدول نگاشت به Markdown"""
181
- if not self.mapping_table:
182
- return "### 📋 جدول نگاشت\n\nهیچ موجودیتی شناسایی نشد"
183
-
184
- table = "### 📋 جدول نگاشت\n\n"
185
- table += "| شناسه | متن اصلی |\n"
186
- table += "|-------|----------|\n"
187
-
188
- for token, original in sorted(self.mapping_table.items()):
189
- table += f"| **{token}** | {original} |\n"
190
-
191
- return table
192
-
193
- # متغیر سراسری
194
- anonymizer = None
195
-
196
- def process(input_text: str, analysis_prompt: str = None, method: str = "cerebras"):
197
- """پردازش متن - 4 مرحله"""
198
- global anonymizer
199
-
200
- if not input_text.strip():
201
- return "", "", "", ""
202
-
203
- cerebras_key = os.getenv("CEREBRAS_API_KEY")
204
- gpt_key = os.getenv("OPENAI_API_KEY")
205
-
206
- if not anonymizer:
207
- anonymizer = AnonymizerAdvanced(cerebras_key, gpt_key)
208
- else:
209
- anonymizer.mapping_table = {}
210
- anonymizer.reverse_mapping = {}
211
-
212
- try:
213
- logger.info("=" * 70)
214
- logger.info(f"🚀 شروع پردازش - روش: Cerebras")
215
- logger.info("=" * 70)
216
-
217
- # ============================================
218
- # مرحله 1: ناشناس‌سازی
219
- # ============================================
220
- logger.info("📝 مرحله 1: ناشناس‌سازی...")
221
-
222
- anonymized_text, _ = anonymizer.anonymize_with_cerebras(input_text)
223
-
224
- logger.info(f"✅ ناشناس‌سازی: {len(anonymized_text)} کاراکتر")
225
-
226
- # ============================================
227
- # مرحله 2: ChatGPT با متن ناشناس‌سازی شده + دستورات
228
- # ============================================
229
- logger.info("🤖 مرحله 2: ChatGPT...")
230
- gpt_response = anonymizer.analyze_with_gpt(anonymized_text, analysis_prompt)
231
- logger.info(f"✅ ChatGPT: {len(gpt_response)} کاراکتر")
232
-
233
- # ============================================
234
- # مرحله 3: بازگردانی پاسخ ChatGPT
235
- # ============================================
236
- logger.info("🔄 مرحله 3: بازگردانی...")
237
- restored_text = anonymizer.restore_text(gpt_response)
238
- logger.info("✅ بازگردانی کامل")
239
-
240
- # ============================================
241
- # مرحله 4: جدول نگاشت
242
- # ============================================
243
- logger.info("📋 مرحله 4: جدول نگاشت...")
244
- mapping_str = anonymizer.get_mapping_table_md()
245
- logger.info(f"✅ {len(anonymizer.mapping_table)} موجودیت")
246
-
247
- logger.info("=" * 70)
248
- logger.info("✅ تمام مراحل کامل!")
249
- logger.info("=" * 70)
250
-
251
- return restored_text, gpt_response, anonymized_text, mapping_str
252
-
253
- except Exception as e:
254
- logger.error(f"❌ خطا: {str(e)}", exc_info=True)
255
- return "", f"❌ خطا: {str(e)}", "", ""
256
-
257
- def clear_all():
258
- """پاک کردن همه"""
259
- return "", "", "", "", "", ""
260
-
261
- # Gradio Interface
262
- css_rtl = """
263
- .input-box { direction: rtl; text-align: right; }
264
- .textbox textarea { direction: rtl; text-align: right; font-family: 'Tahoma', serif; }
265
- """
266
-
267
- with gr.Blocks(title="سیستم ناشناس‌سازی متون", theme=gr.themes.Soft(), css=css_rtl) as app:
268
-
269
- gr.Markdown("# 🔐 سیستم ناشناس‌سازی متون مالی فارسی", elem_classes="input-box")
270
- gr.Markdown("#### استخراج و ناشناس‌سازی موجودیت‌های حساس", elem_classes="input-box")
271
-
272
- # ============================================
273
- # صفحه اول: دکمه‌ها (راست) + ورودی (چپ)
274
- # ============================================
275
- with gr.Row():
276
- # سمت راست: دکمه‌ها و تنظیمات
277
- with gr.Column(scale=1):
278
- gr.Markdown("### ⚙️ تنظیمات", elem_classes="input-box")
279
-
280
- method = gr.Radio(
281
- ["cerebras"],
282
- value="cerebras",
283
- label="🔧 روش پردازش",
284
- elem_classes="input-box"
285
- )
286
-
287
- gr.Markdown("---")
288
-
289
- analysis_prompt = gr.Textbox(
290
- lines=8,
291
- placeholder="دستورات تحلیلی برای ChatGPT...",
292
- label="📋 دستورات تحلیلی",
293
- elem_classes="textbox",
294
- value="خلاصه سازی\nشناسایی موجودیت‌های ناشناس\nنکات کلیدی"
295
- )
296
-
297
- gr.Markdown("---")
298
-
299
- with gr.Column():
300
- process_btn = gr.Button(
301
- "▶️ پردازش",
302
- variant="primary",
303
- size="lg"
304
- )
305
-
306
- clear_btn = gr.Button(
307
- "🗑️ پاک کردن",
308
- variant="stop",
309
- size="lg"
310
- )
311
-
312
- # سمت چپ: متن ورودی (بزرگ‌تر)
313
- with gr.Column(scale=3):
314
- input_text = gr.Textbox(
315
- lines=14,
316
- placeholder="متن مالی/خبری را وارد کنید...",
317
- label="📝 متن ورودی",
318
- elem_classes="textbox"
319
- )
320
-
321
- # ============================================
322
- # صفحه دوم: 3 باکس نتایج (وسط)
323
- # ============================================
324
- gr.Markdown("---")
325
- gr.Markdown("## 📊 نتایج پردازش", elem_classes="input-box")
326
-
327
- with gr.Row():
328
- # باکس 1: متن بازگردانی شده (راست)
329
- with gr.Column(scale=1):
330
- restored_text = gr.Textbox(
331
- lines=12,
332
- label="✅ متن بازگردانی شده",
333
- interactive=False,
334
- elem_classes="textbox"
335
- )
336
-
337
- # باکس 2: تحلیل ChatGPT (وسط)
338
- with gr.Column(scale=1):
339
- gpt_analysis = gr.Textbox(
340
- lines=12,
341
- label="🤖 تحلیل ChatGPT",
342
- interactive=False,
343
- elem_classes="textbox"
344
- )
345
-
346
- # باکس 3: متن ناشناس‌شده (چپ)
347
- with gr.Column(scale=1):
348
- anonymized_text = gr.Textbox(
349
- lines=12,
350
- label="🔒 متن ناشناس‌شده",
351
- interactive=False,
352
- elem_classes="textbox"
353
- )
354
-
355
- # ============================================
356
- # پایین: جدول نگاشت
357
- # ============================================
358
- gr.Markdown("---")
359
-
360
- mapping_table = gr.Textbox(
361
- lines=10,
362
- label="📋 جدول نگاشت",
363
- interactive=False,
364
- elem_classes="textbox"
365
- )
366
-
367
- # ============================================
368
- # Event Handlers
369
- # ============================================
370
- process_btn.click(
371
- fn=process,
372
- inputs=[input_text, analysis_prompt, method],
373
- outputs=[restored_text, gpt_analysis, anonymized_text, mapping_table]
374
- )
375
-
376
- clear_btn.click(
377
- fn=clear_all,
378
- outputs=[input_text, analysis_prompt, restored_text, gpt_analysis, anonymized_text, mapping_table]
379
- )
380
-
381
- if __name__ == "__main__":
382
- print("=" * 70)
383
- print("🚀 سیستم ناشناس‌سازی متون در حال راه‌اندازی...")
384
- print("=" * 70)
385
- print("\n📋 نحوه استفاده:\n")
386
- print("1. CEREBRAS_API_KEY و OPENAI_API_KEY را تنظیم کنید")
387
- print("2. http://localhost:7860 را باز کنید")
388
- print("3. متن را وارد کنید")
389
- print("4. 'پردازش' را کلیک کنید\n")
390
- print("روش استفاده شده: Cerebras (Llama 3.3-70B)")
391
- print("=" * 70 + "\n")
392
-
393
- app.launch(
394
- server_name="0.0.0.0",
395
- server_port=7860,
396
- share=False,
397
- show_error=True
398
- )