Update app.py

app.py

@@ -182,27 +182,23 @@ class AnonymizationEvaluator:
         
         return report
     
-    def create_downloadable_csv(self) -> str:
-        """ایجاد فایل CSV قابل دانلود"""
+    def create_downloadable_csv(self) -> bytes:
+        """ایجاد محتوای CSV برای دانلود مستقیم"""
         if self.results_df is None or self.results_df.empty:
             return None
             
         try:
-            # تولید نام فایل با زمان
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            # تولید محتوای CSV در حافظه
+            csv_buffer = io.StringIO()
+            self.results_df.to_csv(csv_buffer, index=False, encoding='utf-8')
+            csv_content = csv_buffer.getvalue()
+            csv_buffer.close()
             
-            # استفاده از tempfile برای ایجاد فایل موقت
-            with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.csv', 
-                                           prefix=f'evaluation_results_{timestamp}_', 
-                                           encoding='utf-8-sig', newline='') as temp_file:
-                # نوشتن داده‌ها در فایل
-                self.results_df.to_csv(temp_file, index=False)
-                temp_filename = temp_file.name
-            
-            return temp_filename
+            # تبدیل به bytes برای دانلود
+            return csv_content.encode('utf-8-sig')
             
         except Exception as e:
-            print(f"خطا در ایجاد فایل: {str(e)}")
+            print(f"خطا در ایجاد محتوای CSV: {str(e)}")
             return None
 
 def create_evaluation_interface():
@@ -354,15 +350,26 @@ def create_evaluation_interface():
                         gr.File(visible=False)
                     )
                 
-                filename = evaluator.create_downloadable_csv()
-                if filename:
+                # ایجاد محتوای CSV
+                csv_content = evaluator.create_downloadable_csv()
+                if csv_content:
+                    # تولید نام فایل
+                    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+                    filename = f"evaluation_results_{timestamp}.csv"
+                    
+                    # ذخیره در فایل موقت برای دانلود
+                    with tempfile.NamedTemporaryFile(mode='wb', delete=False, 
+                                                   suffix='.csv', prefix='eval_') as temp_file:
+                        temp_file.write(csv_content)
+                        temp_filename = temp_file.name
+                    
                     return (
                         f"✅ فایل نتایج آماده شد: {filename} ({len(evaluator.results_df)} سطر)",
-                        gr.File(value=filename, visible=True)
+                        gr.File(value=temp_filename, visible=True)
                     )
                 else:
                     return (
-                        "❌ خطا در ایجاد فایل دانلود",
+                        "❌ خطا در ایجاد محتوای CSV",
                         gr.File(visible=False)
                     )
             except Exception as e:
@@ -436,19 +443,34 @@ def create_evaluation_interface():
             
             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)
+                try:
+                    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)
+                    
+                    # ایجاد محتوای CSV در حافظه
+                    csv_buffer = io.StringIO()
+                    test_df.to_csv(csv_buffer, index=False)
+                    csv_content = csv_buffer.getvalue()
+                    csv_buffer.close()
+                    
+                    # تبدیل به bytes و ذخیره در فایل موقت
+                    csv_bytes = csv_content.encode('utf-8-sig')
+                    with tempfile.NamedTemporaryFile(mode='wb', delete=False, 
+                                                   suffix='.csv', prefix='test_') as temp_file:
+                        temp_file.write(csv_bytes)
+                        temp_filename = temp_file.name
+                    
+                    return gr.File(value=temp_filename, visible=True)
+                except Exception as e:
+                    print(f"خطا در ایجاد فایل تست: {str(e)}")
+                    return gr.File(visible=False)
             
             test_download_btn.click(
                 fn=create_test_file,