PerSent 0.2.2__tar.gz → 0.2.3__tar.gz

{persent-0.2.2 → persent-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PerSent
-Version: 0.2.2
+Version: 0.2.3
 Summary: Persian Sentiment Analysis Toolkit
 Home-page: https://github.com/RezaGooner/PerSent
 Author: RezaGooner

{persent-0.2.2 → persent-0.2.3}/PerSent/CommentAnalyzer.py

@@ -112,30 +112,38 @@ class CommentAnalyzer:
         self.classifier = joblib.load(os.path.join(self.model_dir, 'classifier.joblib'))
         self.vectorizer = Word2Vec.load(os.path.join(self.model_dir, 'word2vec.model'))
         
-    def csvPredict(self, input_csv, output_path, text_column='body'):
+    def csvPredict(self, input_csv, output_path, text_column=0):
         """
         Analyze sentiment for comments in a CSV file and save results
         
         Parameters:
             input_csv (str): Path to input CSV file
             output_path (str): Path to save output CSV file
-            text_column (str, optional): Name of column containing comments. Defaults to 'body'.
+            text_column (str/int, optional): Name or index (0-based) of column containing comments. 
+                                          Defaults to 0 (first column).
         """
         try:
             # Read input CSV
             df = pd.read_csv(input_csv)
             
-            # Check if specified column exists
-            if text_column not in df.columns:
-                # If only one column exists, use it regardless of name
-                if len(df.columns) == 1:
-                    text_column = df.columns[0]
-                else:
+            # Determine the correct column
+            if isinstance(text_column, int):
+                # Handle negative indices
+                if text_column < 0:
+                    text_column = len(df.columns) + text_column
+                    
+                if text_column >= len(df.columns) or text_column < 0:
+                    raise ValueError(f"Column index {text_column} is out of range")
+                    
+                column_name = df.columns[text_column]
+            else:
+                if text_column not in df.columns:
                     raise ValueError(f"Column '{text_column}' not found in CSV file")
+                column_name = text_column
             
             # Analyze each comment
             tqdm.pandas(desc="Analyzing comments")
-            df['sentiment'] = df[text_column].progress_apply(self.predict)
+            df['sentiment'] = df[column_name].progress_apply(self.predict)
             
             # Save results
             df.to_csv(output_path, index=False, encoding='utf-8-sig')

{persent-0.2.2 → persent-0.2.3}/PerSent/__init__.py

@@ -1,4 +1,4 @@
 from .CommentAnalyzer import CommentAnalyzer
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 __all__ = ['CommentAnalyzer']

{persent-0.2.2 → persent-0.2.3}/PerSent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PerSent
-Version: 0.2.2
+Version: 0.2.3
 Summary: Persian Sentiment Analysis Toolkit
 Home-page: https://github.com/RezaGooner/PerSent
 Author: RezaGooner

{persent-0.2.2 → persent-0.2.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="PerSent",
-    version="0.2.2",
+    version="0.2.3",
     author="RezaGooner",
     author_email="RezaAsadiProgrammer@Gmail.com",
     description="Persian Sentiment Analysis Toolkit",