Update pages/model.py

pages/model.py

@@ -32,26 +32,34 @@ def extract_text_from_pdf(uploaded_file):
         loader = UnstructuredPDFLoader(tmp_path)
         return loader.load()[0].page_content
     except Exception as e:
-        st.error(f"Error reading PDF file: {e}")
+        st.error(f"⚠️ Error reading PDF file: {e}")
         return ""
 
 def extract_text_from_txt(uploaded_file):
     try:
         return uploaded_file.read().decode("utf-8")
     except Exception as e:
-        st.error(f"Error reading text file: {e}")
+        st.error(f"⚠️ Error reading text file: {e}")
         return ""
 
-st.set_page_config(page_title="Resume & JD Extractor", layout="centered")
-st.title("📄 Resume & Job Description Extractor")
+st.set_page_config(page_title="SmartHire AI – Resume & JD Analyzer", layout="centered")
+st.markdown("<h1 style='text-align:center;'>🤖 SmartHire AI</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align:center;color:gray;'>AI-powered Resume & Job Description Analyzer</h4><br>", unsafe_allow_html=True)
 
-resume_file = st.file_uploader("Upload Resume (PDF)", type=["pdf"])
-jd_file = st.file_uploader("Upload Job Description (PDF or TXT)", type=["pdf", "txt"])
-jd_text = st.text_area("Or paste Job Description text here")
+with st.container():
+    st.markdown("### 📂 Upload Center")
+    col1, col2 = st.columns(2)
+    with col1:
+        resume_file = st.file_uploader("Upload Candidate Resume (PDF)", type=["pdf"])
+    with col2:
+        jd_file = st.file_uploader("Upload Job Description (PDF or TXT)", type=["pdf", "txt"])
+    
+    jd_text = st.text_area("📋 Or paste the Job Description text below", height=150)
 
-if st.button("🔍 Extract Data"):
+
+if st.button("🚀 Analyze Now"):
     if not resume_file and not (jd_file or jd_text):
-        st.warning("Please upload at least one file (Resume or JD) or paste JD text.")
+        st.warning("⚠️ Please upload at least one resume or job description.")
     else:
         if resume_file:
             resume_text = extract_text_from_pdf(resume_file)
@@ -62,8 +70,13 @@ if st.button("🔍 Extract Data"):
                     f"Resume:\n{resume_text}"
                 )
                 resume_data = llm.invoke(resume_prompt)
-                st.subheader("📌 Extracted Resume Data")
-                st.markdown(f"<div style='background-color:#f9f9f9;padding:10px;border-radius:8px;'>{resume_data}</div>", unsafe_allow_html=True)
+                st.markdown("---")
+                st.subheader("📘 Candidate Summary")
+                st.markdown(f"""
+                <div style='background-color:#eef6fb;padding:15px;border-left:5px solid #0099cc;border-radius:6px'>
+                    {resume_data}
+                </div>
+                """, unsafe_allow_html=True)
 
         jd_text_extracted = ""
         if jd_file:
@@ -82,5 +95,10 @@ if st.button("🔍 Extract Data"):
                 f"Job Description:\n{jd_text_extracted}"
             )
             jd_data = llm.invoke(jd_prompt)
-            st.subheader("📌 Extracted Job Description Data")
-            st.markdown(f"<div style='background-color:#f9f9f9;padding:10px;border-radius:8px;'>{jd_data}</div>", unsafe_allow_html=True)
+            st.markdown("---")
+            st.subheader("📄 Job Role Breakdown")
+            st.markdown(f"""
+            <div style='background-color:#fff7e6;padding:15px;border-left:5px solid #ffaa00;border-radius:6px'>
+                {jd_data}
+            </div>
+            """, unsafe_allow_html=True)