Update app.py

app.py

@@ -6,6 +6,10 @@ import streamlit as st
 from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
 from io import BytesIO
 
+# Disable SSL verification warning
+import urllib3
+urllib3.disable_warnings()
+
 # Configure page
 st.set_page_config(
     page_title="Resume Scorer",
@@ -13,16 +17,13 @@ st.set_page_config(
     layout="wide"
 )
 
-# Initialize the model and agent
-@st.cache_resource
-def initialize_agent():
-    try:
-        model = HfApiModel()
-        agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
-        return agent
-    except Exception as e:
-        st.error(f"Error initializing agent: {e}")
-        return None
+# Initialize the model and agent - moved outside of cache
+try:
+    model = HfApiModel()
+    agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
+except Exception as e:
+    st.error(f"Error initializing agent: {e}")
+    agent = None
 
 def extract_text_from_uploaded_pdf(pdf_file):
     """Extracts text from an uploaded PDF file."""
@@ -37,8 +38,12 @@ def extract_text_from_uploaded_pdf(pdf_file):
         st.error(f"Error extracting text from PDF: {e}")
         return None
 
-def score_resume_against_job_description(agent, resume_text, job_description):
+def score_resume_against_job_description(resume_text, job_description):
     """Scores a resume against a job description using smolagents."""
+    if agent is None:
+        st.error("Agent not initialized properly")
+        return None
+        
     prompt = f"""
     You are a professional recruiter that is highly skilled in scoring resumes.
     You will receive a resume and a job description.
@@ -55,7 +60,7 @@ def score_resume_against_job_description(agent, resume_text, job_description):
     try:
         response = agent.run(prompt)
         # Ensure we get a numeric response
-        score = int(''.join(filter(str.isdigit, response)))
+        score = int(''.join(filter(str.isdigit, str(response))))
         return min(max(score, 0), 100)  # Ensure score is between 0 and 100
     except Exception as e:
         st.error(f"An error has occurred: {e}")
@@ -66,109 +71,81 @@ def extract_links_from_text(text):
     links = re.findall(r'(https?://\S+)', text)
     return links
 
-# Add custom CSS
-st.markdown("""
-    <style>
-    .main {
-        padding: 2rem;
-    }
-    .stButton>button {
-        width: 100%;
-        margin-top: 1rem;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-# Main UI
-st.title("📄 Resume Scorer")
-st.write("Upload a resume and enter a job description to get a matching score!")
-
-# Initialize agent
-agent = initialize_agent()
-
-if agent is None:
-    st.error("Failed to initialize the scoring system. Please try again later.")
-    st.stop()
-
-# Create two columns
-col1, col2 = st.columns(2)
-
-with col1:
-    st.subheader("Upload Resume")
-    uploaded_file = st.file_uploader("Upload Resume (PDF)", type="pdf")
-
-with col2:
-    st.subheader("Job Description")
-    job_description = st.text_area(
-        "Enter Job Description",
-        height=200,
-        placeholder="Paste the job description here..."
-    )
-
-if uploaded_file and job_description:
-    if st.button("Score Resume"):
-        with st.spinner("Analyzing resume..."):
-            # Extract text from uploaded PDF
-            resume_text = extract_text_from_uploaded_pdf(uploaded_file)
-            
-            if resume_text:
-                # Extract links
-                links = extract_links_from_text(resume_text)
-                
-                # Get score
-                score = score_resume_against_job_description(agent, resume_text, job_description)
-                
-                # Display results
-                st.divider()
-                st.subheader("Results")
+def main():
+    st.title("📄 Resume Scorer")
+    st.write("Upload a resume and enter a job description to get a matching score!")
+
+    if agent is None:
+        st.error("Failed to initialize the scoring system. Please try again later.")
+        return
+
+    # Create two columns
+    col1, col2 = st.columns(2)
+
+    with col1:
+        st.subheader("Upload Resume")
+        uploaded_file = st.file_uploader("Upload Resume (PDF)", type="pdf")
+
+    with col2:
+        st.subheader("Job Description")
+        job_description = st.text_area(
+            "Enter Job Description",
+            height=200,
+            placeholder="Paste the job description here..."
+        )
+
+    if uploaded_file and job_description:
+        if st.button("Score Resume"):
+            with st.spinner("Analyzing resume..."):
+                resume_text = extract_text_from_uploaded_pdf(uploaded_file)
                 
-                if score is not None:
-                    # Create columns for score display
-                    score_col, links_col = st.columns(2)
+                if resume_text:
+                    links = extract_links_from_text(resume_text)
+                    score = score_resume_against_job_description(resume_text, job_description)
                     
-                    with score_col:
-                        st.metric("Resume Match Score", f"{score}/100")
-                        
-                        # Add color-coded feedback
-                        if score >= 80:
-                            st.success("Strong match! 🌟")
-                        elif score >= 60:
-                            st.info("Good match! 👍")
-                        else:
-                            st.warning("Consider revising the resume to better match the job requirements")
+                    st.divider()
+                    st.subheader("Results")
                     
-                    with links_col:
-                        if links:
-                            st.subheader("Links found in Resume:")
-                            for link in links:
-                                st.write(f"- {link}")
-                        else:
-                            st.info("No links found in the resume")
+                    if score is not None:
+                        score_col, links_col = st.columns(2)
+                        
+                        with score_col:
+                            st.metric("Resume Match Score", f"{score}/100")
+                            
+                            if score >= 80:
+                                st.success("Strong match! 🌟")
+                            elif score >= 60:
+                                st.info("Good match! 👍")
+                            else:
+                                st.warning("Consider revising the resume")
+                        
+                        with links_col:
+                            if links:
+                                st.subheader("Links found in Resume:")
+                                for link in links:
+                                    st.write(f"- {link}")
+                            else:
+                                st.info("No links found in the resume")
+                    else:
+                        st.error("Error calculating score")
                 else:
-                    st.error("Error calculating score")
-            else:
-                st.error("Could not extract text from the PDF")
-
-# Add usage instructions in sidebar
-with st.sidebar:
-    st.header("How to Use")
-    st.write("""
-    1. Upload a PDF resume using the file uploader
-    2. Paste the job description in the text area
-    3. Click 'Score Resume' to see the results
-    
-    The score is based on how well the resume matches
-    the job description requirements.
-    """)
-    
-    st.divider()
-    
-    st.markdown("""
-    ### About
-    This tool uses AI to analyze resumes and provide
-    matching scores against job descriptions. The score
-    is based on:
-    - Skills match
-    - Experience relevance
-    - Qualification alignment
-    """)
+                    st.error("Could not extract text from the PDF")
+
+    with st.sidebar:
+        st.header("How to Use")
+        st.write("""
+        1. Upload a PDF resume
+        2. Paste the job description
+        3. Click 'Score Resume'
+        """)
+        
+        st.divider()
+        
+        st.markdown("""
+        ### About
+        This tool uses AI to analyze resumes and provide
+        matching scores against job descriptions.
+        """)
+
+if __name__ == "__main__":
+    main()