Update src/inference.py

src/inference.py

@@ -1,39 +1,45 @@
 import os
-from src.preprocessing import clean_text
-from src.features import create_features
-from src.model import load_model
-from src.entity_extraction import extract_entities
 import numpy as np
 import joblib
 import scipy.sparse
 from textblob import TextBlob
 
-# ✅ Correct path to models folder relative to this file
-BASE_PATH = os.path.join(os.path.dirname(__file__), '..', 'models')
+from src.preprocessing import clean_text
+from src.features import create_features
+from src.model import load_model
+from src.entity_extraction import extract_entities
+
+# ─── Adjusted BASE_PATH to point at ../models ──────────────────────────────────
+BASE_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "models"))
 
-# ✅ Load models
-model_issue = load_model(os.path.join(BASE_PATH, 'model_issue_type.pkl'))
-model_urgency = load_model(os.path.join(BASE_PATH, 'model_urgency_level.pkl'))
-tfidf = joblib.load(os.path.join(BASE_PATH, 'tfidf.pkl'))
+# Load models and TF-IDF vectorizer from the models/ folder at repo root
+model_issue = load_model(os.path.join(BASE_PATH, "model_issue_type.pkl"))
+model_urgency = load_model(os.path.join(BASE_PATH, "model_urgency_level.pkl"))
+tfidf = joblib.load(os.path.join(BASE_PATH, "tfidf.pkl"))
 
 def predict_ticket(ticket_text):
+    # Preprocess
     clean = clean_text(ticket_text)
+    # TF-IDF transform
     X_tfidf = tfidf.transform([clean])
+    # Additional features
     ticket_length = len(clean.split())
     sentiment = TextBlob(clean).sentiment.polarity
 
+    # Combine features (sparse + dense)
     X_features = scipy.sparse.hstack([
         X_tfidf,
         np.array([[ticket_length]]),
         np.array([[sentiment]])
     ])
 
+    # Predictions
     issue_pred = model_issue.predict(X_features)[0]
     urgency_pred = model_urgency.predict(X_features)[0]
     entities = extract_entities(ticket_text)
 
     return {
-        'issue_type': issue_pred,
-        'urgency_level': urgency_pred,
-        'entities': entities
+        "issue_type": issue_pred,
+        "urgency_level": urgency_pred,
+        "entities": entities
     }