Update pages/model.py

pages/model.py

@@ -0,0 +1,53 @@
+import streamlit as st
+import pickle
+import re
+import numpy as np
+
+st.set_page_config(page_title="TagGPT - Stack Overflow Tag Predictor", layout="centered")
+
+def preprocess_input(text):
+    text = re.sub(r"<.*?>", " ", text)                
+    text = re.sub(r"[^a-zA-Z0-9\s]", " ", text)       
+    text = re.sub(r"\s+", " ", text.lower()).strip()  
+    return text
+
+@st.cache_resource
+def load_model_assets():
+    with open("model.pkl", "rb") as f:
+        model = pickle.load(f)
+    with open("tfidf.pkl", "rb") as f:
+        vectorizer = pickle.load(f)
+    with open("mlb.pkl", "rb") as f:
+        encoder = pickle.load(f)
+    return model, vectorizer, encoder
+
+model, vectorizer, encoder = load_model_assets()
+
+st.title("🔖 TagGPT: Predict Stack Overflow Tags")
+st.markdown("Smartly auto-tag your programming questions using ML.")
+
+q_title = st.text_input("📝 Enter Question Title")
+q_body = st.text_area("📄 Describe Your Problem", height=200)
+threshold = st.slider("🎯 Prediction Threshold", 0.1, 0.9, 0.3, 0.05)
+
+if st.button("🔍 Generate Tags"):
+    if not q_title.strip() or not q_body.strip():
+        st.warning("🚨 Please fill in both the title and description.")
+    else:
+        user_query = preprocess_input(q_title + " " + q_body)
+        X_transformed = vectorizer.transform([user_query])
+
+        try:
+            tag_probs = model.predict_proba(X_transformed)
+            tag_preds = (tag_probs >= threshold).astype(int)
+        except AttributeError:
+            st.warning("⚠ Model doesn't support probability output. Falling back to `predict`.")
+            tag_preds = model.predict(X_transformed)
+
+        predicted = encoder.inverse_transform(tag_preds)
+
+        if predicted and predicted[0]:
+            st.success("✅ Predicted Tags:")
+            st.write(", ".join(predicted[0]))
+        else:
+            st.info("🤖 No tags predicted. Try refining your input or lowering the threshold.")