Update model.py

model.py

@@ -13,6 +13,7 @@ import torch
 from transformers import AutoTokenizer, BertForSequenceClassification
 from datasets import Dataset
 import pandas as pd
+import os
 import re
 from hazm import Normalizer, Lemmatizer, word_tokenize, stopwords_list
 
@@ -100,6 +101,11 @@ def process_sentence(sentence):
     return f"Type: {sentence_type}\nSentiment: {sentiment}\nCleaned Text: {cleaned}"
 
 
+# ایجاد پوشه برای ذخیره فایل‌ها در صورت نبود آن
+output_folder = "./outputs"
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
 
 # Function to process file
 def process_file(file):
@@ -113,8 +119,10 @@ def process_file(file):
         df['Type'] = df['Comment'].apply(classify_sentence)
         df['Sentiment'] = df['Comment'].apply(predict_sentiment_labels)
 
-        output_path = "processed_file.csv"
-        df.to_csv(output_path, index=False)
-        return f"File processed successfully! Download it [here](./{output_path})"
+        processed_file_path = os.path.join(output_folder, "processed_file.csv")
+
+        
+        df.to_csv(processed_file_path, index=False)
+        return processed_file_path 
     except Exception as e:
         return str(e)