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 = 1500
+    model: str = "llama-3.1-8b-instant" 
+    max_tokens: int = 1000
     temperature: float = 0.1
 
 class GroqAnonymizer:
@@ -33,37 +33,23 @@ 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, ... جایگزین کنید
+وظیفه: تشخیص و جایگزینی موجودیت‌های حساس:
 
-نکات مهم:
-- اگر همان موجودیت تکرار شود، همان شماره را استفاده کنید
-- پیشوندهای افراد (دکتر، مهندس، آقا، خانم) را حفظ کنید
-- فقط موجودیت‌های حساس را تغییر دهید، بقیه متن دست نخورده باقی بماند
-- خروجی را فقط به صورت JSON با فیلدهای زیر ارائه دهید:
-{
-  "anonymized_text": "متن ناشناس‌سازی شده",
-  "entities": [
-    {
-      "original": "متن اصلی",
-      "anonymized": "متن جایگزین", 
-      "type": "نوع موجودیت"
-    }
-  ],
-  "statistics": {
-    "company": تعداد,
-    "person": تعداد,
-    "amount": تعداد,
-    "percent": تعداد,
-    "group": تعداد
-  }
-}"""
+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 درآمد دارد"""
 
     def _make_api_request(self, text: str) -> Dict[str, Any]:
         """ارسال درخواست به Groq API"""
@@ -80,7 +66,7 @@ class GroqAnonymizer:
                 },
                 {
                     "role": "user", 
-                    "content": f"لطفاً متن زیر را ناشناس‌سازی کنید:\n\n{text}"
+                    "content": text
                 }
             ],
             "model": self.config.model,
@@ -120,38 +106,53 @@ class GroqAnonymizer:
             
             content = response["choices"][0]["message"]["content"]
             
-            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 قابل پارس نبود"
-                }
+            # پاک کردن 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", {})
+            }
                 
         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"""
@@ -201,19 +202,28 @@ def create_interface():
         این سیستم قادر است نام افراد، شرکت‌ها، مبالغ مالی و درصدها را به صورت هوشمند تشخیص و ناشناس‌سازی کند.
         """)
         
-        # نمایش وضعیت API
-        if api_key_available:
+        # ورود کلید API اگر در secrets موجود نباشد
+        if not api_key_available:
             gr.Markdown("""
-            <div class="success-box">
-            ✅ <strong>سیستم آماده است</strong> - کلید API تنظیم شده
+            <div class="warning-box">
+            ⚠️ <strong>کلید API در متغیرهای محیطی یافت نشد</strong><br>
+            لطفاً کلید Groq API خود را در زیر وارد کنید یا در Settings > Secrets تنظیم کنید
             </div>
             """)
+            
+            api_key_input = gr.Textbox(
+                label="🔑 کلید Groq API",
+                placeholder="gsk_...",
+                type="password",
+                value=""
+            )
         else:
             gr.Markdown("""
-            <div class="warning-box">
-            ⚠️ <strong>کلید API تنظیم نشده</strong> - برای استفاده نیاز به تنظیم GROQ_API_KEY در متغیرهای محیطی
+            <div class="success-box">
+            ✅ <strong>سیستم آماده است</strong> - کلید API تنظیم شده
             </div>
             """)
+            api_key_input = gr.Textbox(visible=False)
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -228,8 +238,7 @@ def create_interface():
                     anonymize_btn = gr.Button(
                         "🔒 ناشناس‌سازی متن", 
                         variant="primary", 
-                        size="lg",
-                        interactive=api_key_available
+                        size="lg"
                     )
                     clear_btn = gr.Button(
                         "🗑️ پاک کردن", 
@@ -243,6 +252,22 @@ def create_interface():
                     max_lines=20,
                     elem_classes=["result-box"]
                 )
+                
+                # دکمه کپی
+                copy_btn = gr.Button(
+                    "📋 کپی متن",
+                    variant="secondary",
+                    size="sm"
+                )
+        
+        # متن برای کپی
+        copy_output = gr.Textbox(
+            label="📋 متن برای کپی (Ctrl+A و Ctrl+C)",
+            lines=3,
+            max_lines=10,
+            visible=False,
+            interactive=True
+        )
         
         with gr.Row():
             with gr.Column():
@@ -252,12 +277,15 @@ def create_interface():
         
         entities_output = gr.Markdown(label="📋 جزئیات تغییرات")
         
-        def process_text(text: str):
+        def process_text(text: str, api_key_manual: str = ""):
             """پردازش متن"""
-            if not api_key_available:
+            # تعیین کلید 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:
                 return (
                     "",
-                    "❌ کلید API تنظیم نشده است",
+                    "❌ کلید API وارد نشده است. لطفاً در فیلد بالا وارد کنید یا در Settings تنظیم کنید.",
                     "",
                     ""
                 )
@@ -271,7 +299,7 @@ def create_interface():
                 )
             
             try:
-                anonymizer = GroqAnonymizer()
+                anonymizer = GroqAnonymizer(api_key=final_api_key)
                 result = anonymizer.anonymize_text(text)
                 
                 if not result["success"]:
@@ -310,7 +338,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", [])
@@ -339,20 +367,28 @@ def create_interface():
                     ""
                 )
         
+        def copy_text(text_to_copy):
+            """تابع کپی متن"""
+            if not text_to_copy.strip():
+                return gr.Textbox(visible=False), "⚠️ متنی برای کپی وجود ندارد"
+            
+            # نمایش textbox با محتوای قابل کپی
+            return gr.Textbox(value=text_to_copy, visible=True), "✅ متن در کادر زیر آماده کپی است (Ctrl+A سپس Ctrl+C)"
+        
         def clear_all():
             """پاک کردن تمام فیلدها"""
-            return "", "", "", "", ""
+            return "", "", "", "", "", gr.Textbox(visible=False)
         
         # اتصال رویدادها
         anonymize_btn.click(
             fn=process_text,
-            inputs=[input_text],
+            inputs=[input_text, api_key_input],
             outputs=[output_text, statistics_output, usage_output, entities_output]
         )
         
         clear_btn.click(
             fn=clear_all,
-            outputs=[input_text, output_text, statistics_output, usage_output, entities_output]
+            outputs=[input_text, output_text, statistics_output, usage_output, entities_output, copy_output]
         )
         
         # مثال‌های نمونه