Update app.py

app.py

@@ -1,21 +1,16 @@
-import pickle
 import streamlit as st
-import os
+import pickle
 import numpy as np
 
-# 💡 Define the custom tokenizer exactly as used during training
-def custom_tokenizer(text):
-    # Modify this function to match your original tokenizer logic
-    return text.lower().split()
-
-
+# ⚙️ Set Streamlit Page Config (this MUST be first)
+st.set_page_config(page_title="Multi-Tag Stack Overflow Predictor", page_icon="🧠")
 
-# 🔃 Load model files
+# 🧪 Load Model Files
 try:
     with open("vectorizer.pkl", "rb") as f:
         vectorizer = pickle.load(f)
 
-    with open("model (2).pkl", "rb") as f:
+    with open("model.pkl", "rb") as f:
         model = pickle.load(f)
 
     with open("binarizer.pkl", "rb") as f:
@@ -25,33 +20,40 @@ except Exception as e:
     st.error(f"❌ Error loading model files: {str(e)}")
     st.stop()
 
-# 🧠 Prediction function
-def predict_tags(title, description):
+# 🧠 Prediction Function (Multi-tag with Threshold)
+def predict_tags(title, description, threshold=0.3):
     try:
         if not title.strip() or not description.strip():
             return "⚠️ Please enter both title and description."
 
         input_text = title + " " + description
         input_vector = vectorizer.transform([input_text])
-        prediction = model.predict(input_vector)
-        predicted_tags = mlb.inverse_transform(prediction)
-        st.write(predicted_tags)
+
+        # Predict probabilities
+        probas = model.predict_proba(input_vector)
+
+        # Apply threshold to get binary predictions
+        multi_pred = (probas >= threshold).astype(int)
+
+        # Convert binary predictions to tags
+        predicted_tags = mlb.inverse_transform(multi_pred)
 
         if predicted_tags and predicted_tags[0]:
-            return "✅ Predicted Tags: " + ", ".join(predicted_tags[0])
+            return "✅ **Predicted Tags:** " + ", ".join(predicted_tags[0])
         else:
-            return "ℹ️ No tags predicted. Try refining your question."
+            return "ℹ️ No tags predicted. Try lowering the threshold or refining your input."
 
     except Exception as e:
         return f"❌ Error during prediction: {str(e)}"
 
 # 🚀 Streamlit UI
-st.title("🔖:red[ Stack Overflow Tags Predictor]")
-st.markdown(":blue[Enter a question title and description to predict relevant tags.]")
+st.title("🔖 Stack Overflow Multi-Tag Predictor")
+st.markdown("Enter a question title and description to predict relevant tags using ML.")
 
-title = st.text_input("📌 Enter Question Title")
-description = st.text_area("📝 Enter Question Description", height=150)
+title = st.text_input("📌 Question Title")
+description = st.text_area("📝 Question Description", height=150)
+threshold = st.slider("🎯 Prediction Threshold", 0.1, 0.9, 0.3, step=0.05)
 
 if st.button("Predict Tags"):
-    result = predict_tags(title, description)
-    st.markdown(result)
+    result = predict_tags(title, description, threshold)
+    st.markdown(result)