Update src/app.py

src/app.py

@@ -570,6 +570,19 @@ def main():
     # Initialize analyzer
     try:
         analyzer = ResumeAnalyzer()
+        # Add ML/AI Engineer to job keywords
+        analyzer.job_keywords["Machine Learning Engineer"] = [
+            "python", "tensorflow", "pytorch", "scikit-learn", "pandas", "numpy", "machine learning",
+            "deep learning", "neural networks", "computer vision", "nlp", "data science", "algorithms",
+            "statistics", "linear algebra", "calculus", "regression", "classification", "clustering",
+            "feature engineering", "model deployment", "mlops", "docker", "kubernetes", "aws", "gcp"
+        ]
+        analyzer.job_keywords["AI Engineer"] = [
+            "artificial intelligence", "machine learning", "deep learning", "neural networks", "python",
+            "tensorflow", "pytorch", "computer vision", "nlp", "natural language processing", "opencv",
+            "transformers", "bert", "gpt", "reinforcement learning", "generative ai", "llm", "chatbot",
+            "model optimization", "ai ethics", "edge ai", "quantization", "onnx", "tensorrt"
+        ]
     except Exception as e:
         st.error(f"Error initializing analyzer: {str(e)}")
         return
@@ -584,6 +597,8 @@ def main():
         st.session_state.chat_history = []
     if "resume_context" not in st.session_state:
         st.session_state.resume_context = ""
+    if "show_chat" not in st.session_state:
+        st.session_state.show_chat = False
     
     # File upload section
     st.header("📁 Upload Your Resume")
@@ -624,9 +639,9 @@ def main():
                 ats_score = analyzer.calculate_ats_score(text, sections)
                 
                 # Create tabs for different analyses
-                tab1, tab2, tab3, tab4, tab5, tab6 = st.tabs([
+                tab1, tab2, tab3, tab4, tab5 = st.tabs([
                     "📊 Overview", "🎯 Skills Analysis", "📝 Section Breakdown", 
-                    "🔍 ATS Analysis", "📋 Report & Suggestions", "🤖 AI Assistant"
+                    "🔍 ATS Analysis", "📋 Report & Suggestions"
                 ])
                 
                 with tab1:
@@ -699,6 +714,27 @@ def main():
                                            title='Technical Skills Found')
                                 fig.update_xaxis(tickangle=45)
                                 st.plotly_chart(fig, use_container_width=True)
+                        else:
+                            st.warning("No technical skills detected")
+                            st.info("💡 Consider adding technical skills relevant to your field")
+                    
+                    with col2:
+                        st.subheader("🤝 Soft Skills")
+                        if soft_skills:
+                            skills_text = " • ".join(soft_skills)
+                            st.success(f"Found {len(soft_skills)} soft skills:")
+                            st.write(skills_text)
+                            
+                            if len(soft_skills) > 3:
+                                soft_df = pd.DataFrame({
+                                    'Skill': soft_skills[:8],
+                                    'Count': [1] * len(soft_skills[:8])
+                                })
+                                fig = px.bar(soft_df, x='Skill', y='Count', 
+                                           title='Soft Skills Found',
+                                           color='Skill')
+                                fig.update_xaxis(tickangle=45)
+                                st.plotly_chart(fig, use_container_width=True)
                         else:
                             st.warning("No soft skills detected")
                             st.info("💡 Consider highlighting soft skills like leadership, communication, teamwork")
@@ -873,63 +909,70 @@ def main():
                         except Exception as e:
                             st.error(f"Error generating PDF: {str(e)}")
                 
-                with tab6:
-                    st.header("🤖 AI Assistant Chat")
-                    
+                # AI Assistant Chat Section (Outside of tabs)
+                st.markdown("---")
+                st.header("🤖 AI Assistant Chat")
+                
+                # Toggle chat visibility
+                if st.button("💬 Toggle AI Assistant Chat"):
+                    st.session_state.show_chat = not st.session_state.show_chat
+                
+                if st.session_state.show_chat:
                     if not os.getenv('OPENROUTER_API_KEY'):
                         st.error("Claude API key not configured. Please set OPENROUTER_API_KEY environment variable.")
-                        return
-                    
-                    # Display chat history
-                    for chat in st.session_state.chat_history:
-                        with st.chat_message(chat["role"]):
-                            st.markdown(chat["content"])
-                    
-                    # Chat input
-                    if prompt := st.chat_input("Ask me anything about your resume..."):
-                        # Add user message to chat history
-                        st.session_state.chat_history.append({"role": "user", "content": prompt})
+                    else:
+                        # Display chat history
+                        chat_container = st.container()
+                        with chat_container:
+                            for chat in st.session_state.chat_history:
+                                with st.chat_message(chat["role"]):
+                                    st.markdown(chat["content"])
                         
-                        # Display user message
-                        with st.chat_message("user"):
-                            st.markdown(prompt)
+                        # Suggested questions
+                        st.subheader("💡 Suggested Questions")
+                        col1, col2, col3 = st.columns(3)
                         
-                        # Generate AI response
-                        with st.chat_message("assistant"):
-                            with st.spinner("Thinking..."):
-                                response = analyzer.chatbot.generate_response(
-                                    prompt, 
-                                    st.session_state.resume_context
-                                )
-                            st.markdown(response)
+                        with col1:
+                            if st.button("How can I improve my resume?"):
+                                prompt = "How can I improve my resume?"
+                                st.session_state.chat_history.append({"role": "user", "content": prompt})
+                                
+                                with st.spinner("Thinking..."):
+                                    response = analyzer.chatbot.generate_response(prompt, st.session_state.resume_context)
+                                
+                                st.session_state.chat_history.append({"role": "assistant", "content": response})
+                                st.rerun()
                         
-                        # Add assistant response to chat history
-                        st.session_state.chat_history.append({"role": "assistant", "content": response})
-                    
-                    # Suggested questions
-                    st.subheader("💡 Suggested Questions")
-                    col1, col2 = st.columns(2)
-                    
-                    with col1:
-                        if st.button("How can I improve my resume?"):
-                            st.session_state.chat_history.append({
-                                "role": "user", 
-                                "content": "How can I improve my resume?"
-                            })
-                            st.experimental_rerun()
-                    
-                    with col2:
-                        if st.button("What skills should I add?"):
-                            st.session_state.chat_history.append({
-                                "role": "user", 
-                                "content": f"What skills should I add for a {selected_role} position?"
-                            })
-                            st.experimental_rerun()
-                    
-                    # Clear chat button
-                    if st.button("🗑️ Clear Chat History"):
-                        st.session_state.chat_history = []
-                        st.experimental_rerun()
+                        with col2:
+                            if st.button("What skills should I add?"):
+                                prompt = f"What skills should I add for a {selected_role} position?"
+                                st.session_state.chat_history.append({"role": "user", "content": prompt})
+                                
+                                with st.spinner("Thinking..."):
+                                    response = analyzer.chatbot.generate_response(prompt, st.session_state.resume_context)
+                                
+                                st.session_state.chat_history.append({"role": "assistant", "content": response})
+                                st.rerun()
+                        
+                        with col3:
+                            if st.button("🗑️ Clear Chat History"):
+                                st.session_state.chat_history = []
+                                st.rerun()
+                        
+                        # Chat input (outside of tabs)
+                        if prompt := st.chat_input("Ask me anything about your resume..."):
+                            # Add user message to chat history
+                            st.session_state.chat_history.append({"role": "user", "content": prompt})
+                            
+                            # Generate AI response
+                            with st.spinner("Thinking..."):
+                                response = analyzer.chatbot.generate_response(prompt, st.session_state.resume_context)
+                            
+                            # Add assistant response to chat history
+                            st.session_state.chat_history.append({"role": "assistant", "content": response})
+                            
+                            # Rerun to show new messages
+                            st.rerun()
             
             except Exception as e:
                 st.error(f"Error during analysis: {str(e)}")