Update app.py

app.py

@@ -189,10 +189,13 @@ class AnonymizationEvaluator:
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         filename = f"evaluation_results_{timestamp}.csv"
         
-        # ذخیره فایل
-        self.results_df.to_csv(filename, index=False, encoding='utf-8-sig')
-        
-        return filename
+        try:
+            # ذخیره فایل با encoding مناسب برای فارسی
+            self.results_df.to_csv(filename, index=False, encoding='utf-8-sig')
+            return filename
+        except Exception as e:
+            print(f"خطا در ایجاد فایل: {str(e)}")
+            return None
 
 def create_evaluation_interface():
     """ایجاد رابط کاربری ارزیابی"""
@@ -254,7 +257,7 @@ def create_evaluation_interface():
                 )
                 
                 download_btn = gr.Button(
-                    "💾 دانلود نتایج",
+                    "💾 دانلود نتایج CSV",
                     variant="secondary",
                     visible=False
                 )
@@ -337,10 +340,16 @@ def create_evaluation_interface():
         
         def download_results():
             try:
+                if evaluator.results_df is None or evaluator.results_df.empty:
+                    return (
+                        "❌ هیچ داده‌ای برای دانلود وجود ندارد. ابتدا ارزیابی را انجام دهید.",
+                        gr.File(visible=False)
+                    )
+                
                 filename = evaluator.create_downloadable_csv()
                 if filename:
                     return (
-                        "✅ فایل نتایج آماده دانلود است",
+                        f"✅ فایل نتایج آماده شد: {filename} ({len(evaluator.results_df)} سطر)",
                         gr.File(value=filename, visible=True)
                     )
                 else:
@@ -402,7 +411,41 @@ def create_evaluation_interface():
             - فایل خروجی شامل ستون‌های اصلی + سه ستون متریک خواهد بود
             - متریک‌ها برای هر سطر جداگانه محاسبه می‌شوند
             - آمار کلی در گزارش خلاصه نمایش داده می‌شود
+            
+            ### مشکل در دانلود؟
+            
+            اگر فایل دانلود نمی‌شود:
+            1. مرورگر خود را رفرش کنید
+            2. مجدداً ارزیابی را انجام دهید
+            3. اطمینان حاصل کنید که popup blocker غیرفعال است
             """)
+        
+        # تست دانلود
+        with gr.Accordion("🧪 تست دانلود", open=False):
+            gr.Markdown("برای تست عملکرد دانلود:")
+            test_download_btn = gr.Button("تست دانلود فایل نمونه")
+            test_file_output = gr.File(label="فایل تست", visible=False)
+            
+            def create_test_file():
+                """ایجاد فایل تست برای بررسی دانلود"""
+                import pandas as pd
+                test_data = {
+                    'original_text': ['متن تست 1', 'متن تست 2'],
+                    'Reference_text': ['company-01 amount-01', 'person-01 amount-02'],
+                    'anonymized_text': ['company-01 amount-01', 'person-01 amount-02'],
+                    'Precision': [1.0, 1.0],
+                    'Recall': [1.0, 1.0],
+                    'F1_Score': [1.0, 1.0]
+                }
+                test_df = pd.DataFrame(test_data)
+                test_filename = "test_download.csv"
+                test_df.to_csv(test_filename, index=False, encoding='utf-8-sig')
+                return gr.File(value=test_filename, visible=True)
+            
+            test_download_btn.click(
+                fn=create_test_file,
+                outputs=[test_file_output]
+            )
     
     return interface