Update app.py

app.py

@@ -1,32 +1,36 @@
-import numpy as np
-import pandas as pd
-import streamlit as st
 import pickle
+import streamlit as st
+import os
 
-# Load models using pickle
+# Show current working directory and files for debugging
+st.write("📂 Current directory:", os.getcwd())
+st.write("📁 Files found:", os.listdir())
+
+# Load the pickle files safely
 try:
-    with open("tfidf_vectorizer(1).pkl", "rb") as f:
+    with open("tfidf_vectorizer.pkl", "rb") as f:
         vectorizer = pickle.load(f)
-except FileNotFoundError:
-    st.error("❌ 'tfidf_vectorizer(1).pkl' not found. Please make sure the file is uploaded to the root directory.")
-    st.stop()
-with open("model(1).pkl", "rb") as f:
-    model = pickle.load(f)
 
-with open("mlb.pkl", "rb") as f:
-    mlb = pickle.load(f)
+    with open("model.pkl", "rb") as f:
+        model = pickle.load(f)
 
-# Streamlit App UI
-st.set_page_config(page_title="🔖 Stack Overflow Tags Predictor", layout="centered")
+    with open("mlb.pkl", "rb") as f:
+        mlb = pickle.load(f)
 
+except FileNotFoundError as e:
+    st.error(f"❌ File not found: {e.filename}")
+    st.stop()
+except Exception as e:
+    st.error(f"❌ Error loading model files: {str(e)}")
+    st.stop()
+
+# Streamlit UI
 st.title("🔖 Stack Overflow Tags Predictor")
-st.write("Enter the question's title and description to get suggested tags.")
+st.markdown("Enter a Stack Overflow question title and description to get predicted tags.")
 
-# Input fields
-title = st.text_input("Enter Question Title:")
-description = st.text_area("Enter Question Description:")
+title = st.text_input("Enter Question Title")
+description = st.text_area("Enter Question Description", height=150)
 
-# Prediction logic
 if st.button("Predict Tags"):
     if not title.strip() or not description.strip():
         st.warning("⚠️ Please enter both title and description.")
@@ -41,5 +45,6 @@ if st.button("Predict Tags"):
                 st.success("✅ Predicted Tags: " + ", ".join(predicted_tags[0]))
             else:
                 st.info("ℹ️ No tags predicted. Try refining your question.")
+
         except Exception as e:
-            st.error(f"❌ Error: {str(e)}")
+            st.error(f"❌ Error during prediction: {str(e)}")