Update app.py

app.py

@@ -14,8 +14,8 @@ class GroqConfig:
     """تنظیمات Groq API"""
     api_key: str
     base_url: str = "https://api.groq.com/openai/v1"
-    model: str = "llama-3.1-8b-instant" 
-    max_tokens: int = 1000
+    model: str = "llama-3.1-8b-instant"
+    max_tokens: int = 1500
     temperature: float = 0.1
 
 class GroqAnonymizer:
@@ -33,23 +33,37 @@ class GroqAnonymizer:
     
     def _create_system_prompt(self) -> str:
         """ایجاد دستورالعمل سیستمی برای Groq"""
-        return """شما یک سیستم ناشناس‌سازی متن فارسی هستید.
+        return """شما یک سیستم ناشناس‌سازی متن فارسی هستید. وظیفه شما تشخیص و جایگزینی موجودیت‌های حساس در متن است.
 
-وظیفه: تشخیص و جایگزینی موجودیت‌های حساس:
+قوانین ناشناس‌سازی:
+1. نام شرکت‌ها را با company-01, company-02, ... جایگزین کنید
+2. نام افراد را با person-01, person-02, ... جایگزین کنید  
+3. مبالغ و اعداد را با amount-01, amount-02, ... جایگزین کنید
+4. درصدها را با percent-01, percent-02, ... جایگزین کنید
+5. نام گروه‌ها را با group-01, group-02, ... جایگزین کنید
 
-1. نام شرکت‌ها → company-01, company-02, ...
-2. نام افراد → person-01, person-02, ...  
-3. مبالغ و اعداد → amount-01, amount-02, ...
-4. درصدها → percent-01, percent-02, ...
-
-نکات:
-- همان موجودیت = همان شماره
-- پیشوندها (دکتر، آقا) را حفظ کنید
-- فقط متن ناشناس‌سازی شده را برگردانید، بدون JSON یا توضیح اضافی
-
-مثال:
-ورودی: احمد رضایی مدیرعامل شرکت پارس 100 میلیون تومان درآمد دارد
-خروجی: person-01 مدیرعامل company-01 amount-01 درآمد دارد"""
+نکات مهم:
+- اگر همان موجودیت تکرار شود، همان شماره را استفاده کنید
+- پیشوندهای افراد (دکتر، مهندس، آقا، خانم) را حفظ کنید
+- فقط موجودیت‌های حساس را تغییر دهید، بقیه متن دست نخورده باقی بماند
+- خروجی را فقط به صورت JSON با فیلدهای زیر ارائه دهید:
+{
+  "anonymized_text": "متن ناشناس‌سازی شده",
+  "entities": [
+    {
+      "original": "متن اصلی",
+      "anonymized": "متن جایگزین", 
+      "type": "نوع موجودیت"
+    }
+  ],
+  "statistics": {
+    "company": تعداد,
+    "person": تعداد,
+    "amount": تعداد,
+    "percent": تعداد,
+    "group": تعداد
+  }
+}"""
 
     def _make_api_request(self, text: str) -> Dict[str, Any]:
         """ارسال درخواست به Groq API"""
@@ -106,53 +120,38 @@ class GroqAnonymizer:
             
             content = response["choices"][0]["message"]["content"]
             
-            # پاک کردن markdown اگر وجود دارد
-            if "```" in content:
-                lines = content.split('\n')
-                clean_lines = []
-                skip = False
-                for line in lines:
-                    if line.strip().startswith('```'):
-                        skip = not skip
-                        continue
-                    if not skip:
-                        clean_lines.append(line)
-                content = '\n'.join(clean_lines)
-            
-            # حذف خطوط اضافی و فضاهای خالی
-            content = content.strip()
-            
-            return {
-                "success": True,
-                "anonymized_text": content,
-                "entities": [],  # در حالت ساده entities نداریم
-                "statistics": self._count_entities(content),
-                "usage": response.get("usage", {})
-            }
+            try:
+                # حذف markdown formatting
+                if "```json" in content:
+                    content = content.split("```json")[1].split("```")[0]
+                elif "```" in content:
+                    content = content.split("```")[1].split("```")[0]
+                
+                result = json.loads(content.strip())
+                
+                return {
+                    "success": True,
+                    "anonymized_text": result.get("anonymized_text", ""),
+                    "entities": result.get("entities", []),
+                    "statistics": result.get("statistics", {}),
+                    "usage": response.get("usage", {})
+                }
+                
+            except json.JSONDecodeError:
+                return {
+                    "success": True,
+                    "anonymized_text": content,
+                    "entities": [],
+                    "statistics": {},
+                    "usage": response.get("usage", {}),
+                    "note": "��اسخ به صورت JSON قابل پارس نبود"
+                }
                 
         except Exception as e:
             return {
                 "success": False,
                 "error": f"خطا در پردازش: {str(e)}"
             }
-    
-    def _count_entities(self, text: str) -> Dict[str, int]:
-        """شمارش موجودیت‌ها در متن ناشناس‌سازی شده"""
-        import re
-        
-        company_count = len(re.findall(r'company-\d+', text))
-        person_count = len(re.findall(r'person-\d+', text))
-        amount_count = len(re.findall(r'amount-\d+', text))
-        percent_count = len(re.findall(r'percent-\d+', text))
-        group_count = len(re.findall(r'group-\d+', text))
-        
-        return {
-            "company": company_count,
-            "person": person_count,
-            "amount": amount_count,
-            "percent": percent_count,
-            "group": group_count
-        }
 
 def create_interface():
     """ایجاد رابط کاربری برای Hugging Face Spaces"""
@@ -202,28 +201,19 @@ def create_interface():
         این سیستم قادر است نام افراد، شرکت‌ها، مبالغ مالی و درصدها را به صورت هوشمند تشخیص و ناشناس‌سازی کند.
         """)
         
-        # ورود کلید API اگر در secrets موجود نباشد
-        if not api_key_available:
+        # نمایش وضعیت API
+        if api_key_available:
             gr.Markdown("""
-            <div class="warning-box">
-            ⚠️ <strong>کلید API در متغیرهای محیطی یافت نشد</strong><br>
-            لطفاً کلید Groq API خود را در زیر وارد کنید یا در Settings > Secrets تنظیم کنید
+            <div class="success-box">
+            ✅ <strong>سیستم آماده است</strong> - کلید API تنظیم شده
             </div>
             """)
-            
-            api_key_input = gr.Textbox(
-                label="🔑 کلید Groq API",
-                placeholder="gsk_...",
-                type="password",
-                value=""
-            )
         else:
             gr.Markdown("""
-            <div class="success-box">
-            ✅ <strong>سیستم آماده است</strong> - کلید API تنظیم شده
+            <div class="warning-box">
+            ⚠️ <strong>کلید API تنظیم نشده</strong> - برای استفاده نیاز به تنظیم GROQ_API_KEY در متغیرهای محیطی
             </div>
             """)
-            api_key_input = gr.Textbox(visible=False)
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -238,7 +228,8 @@ def create_interface():
                     anonymize_btn = gr.Button(
                         "🔒 ناشناس‌سازی متن", 
                         variant="primary", 
-                        size="lg"
+                        size="lg",
+                        interactive=api_key_available
                     )
                     clear_btn = gr.Button(
                         "🗑️ پاک کردن", 
@@ -261,15 +252,12 @@ def create_interface():
         
         entities_output = gr.Markdown(label="📋 جزئیات تغییرات")
         
-        def process_text(text: str, api_key_manual: str = ""):
+        def process_text(text: str):
             """پردازش متن"""
-            # تعیین کلید API - اولویت با manual input
-            final_api_key = api_key_manual.strip() if api_key_manual.strip() else os.getenv("GROQ_API_KEY")
-            
-            if not final_api_key:
+            if not api_key_available:
                 return (
                     "",
-                    "❌ کلید API وارد نشده است. لطفاً در فیلد بالا وارد کنید یا در Settings تنظیم کنید.",
+                    "❌ کلید API تنظیم نشده است",
                     "",
                     ""
                 )
@@ -283,7 +271,7 @@ def create_interface():
                 )
             
             try:
-                anonymizer = GroqAnonymizer(api_key=final_api_key)
+                anonymizer = GroqAnonymizer()
                 result = anonymizer.anonymize_text(text)
                 
                 if not result["success"]:
@@ -322,7 +310,7 @@ def create_interface():
                     usage_md += f"• Token های ورودی: {usage.get('prompt_tokens', 'نامشخص')}\n"
                     usage_md += f"• Token های خروجی: {usage.get('completion_tokens', 'نامشخص')}\n"
                 else:
-                    usage_md += "✅ پردازش با موفقیت انجام شد"
+                    usage_md += "اطلاعات در دسترس نیست"
                 
                 # جزئیات موجودیت‌ها
                 entities = result.get("entities", [])
@@ -358,7 +346,7 @@ def create_interface():
         # اتصال رویدادها
         anonymize_btn.click(
             fn=process_text,
-            inputs=[input_text, api_key_input],
+            inputs=[input_text],
             outputs=[output_text, statistics_output, usage_output, entities_output]
         )