Update data_processor.py

data_processor.py

@@ -233,7 +233,8 @@
 import re
 import pandas as pd
 import os
-from huggingface_hub import InferenceClient
+# from huggingface_hub import InferenceClient
+from mistralai import Mistral     
 
 class DataProcessor:
     INTERVENTION_COLUMN_OPTIONS = [
@@ -245,11 +246,21 @@ class DataProcessor:
     PARTIALLY_ENGAGED_STR = 'Partially Engaged'
     NOT_ENGAGED_STR = 'Not Engaged'
 
+    # HF INFERENCE
+    # def __init__(self, student_metrics_df=None):
+    #     self.hf_api_key = os.getenv('HF_API_KEY')
+    #     if not self.hf_api_key:
+    #         raise ValueError("HF_API_KEY not set in environment variables")
+    #     self.client = InferenceClient(api_key=self.hf_api_key)
+    #     self.student_metrics_df = student_metrics_df
+    #     self.intervention_column = None  # Will be set when processing data
+
+    # MISTRAL INFERENCE
     def __init__(self, student_metrics_df=None):
-        self.hf_api_key = os.getenv('HF_API_KEY')
-        if not self.hf_api_key:
-            raise ValueError("HF_API_KEY not set in environment variables")
-        self.client = InferenceClient(api_key=self.hf_api_key)
+        self.mistral_api_key = os.getenv('MISTRAL_API_KEY')
+        if not self.mistral_api_key:
+            raise ValueError("MISTRAL_API_KEY not set in environment variables")
+        self.client = Mistral(api_key=self.mistral_api_key)
         self.student_metrics_df = student_metrics_df
         self.intervention_column = None  # Will be set when processing data