Update src/app.py

src/app.py

@@ -60,51 +60,35 @@ from reportlab.lib.styles import getSampleStyleSheet
 from reportlab.lib.units import inch
 
 # Claude Chatbot Class
+import time
+
 class ClaudeChatbot:
     def __init__(self):
         self.api_key = os.getenv('OPENROUTER_API_KEY')
         self.base_url = "https://openrouter.ai/api/v1/chat/completions"
         
-        # Using free models on OpenRouter
-        self.model = "meta-llama/llama-3.2-3b-instruct:free"
+        self.model = "meta-llama/llama-3.2-3b-instruct:free"  
         
         if not self.api_key:
             st.error("❌ OPENROUTER_API_KEY not found in environment variables!")
             st.info("Please set your OpenRouter API key in the environment variables.")
     
     def generate_response(self, prompt, context="", max_tokens=1500):
-        """Generate response using free models via OpenRouter - No rate limiting"""
+        """Generate response using free models via OpenRouter"""
         if not self.api_key:
             return "Error: API key not configured"
         
         headers = {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json",
-            "HTTP-Referer": "https://your-app-url.com",
+            "HTTP-Referer": "https://your-app-url.com",  # Replace with your actual URL
             "X-Title": "AI Resume Analyzer"
         }
         
-        # Enhanced system prompt for resume-focused responses
-        system_prompt = """You are an expert resume and career consultant with deep knowledge of hiring practices, ATS systems, and industry requirements. 
-
-        IMPORTANT GUIDELINES:
-        - Provide specific, actionable advice for resume improvement
-        - Keep responses focused on resume optimization and career advice
-        - Be encouraging but honest in feedback
-        - Limit responses to resume-related topics only
-        - If asked about non-resume topics, politely redirect to resume discussion
-        - Provide concrete examples when suggesting improvements
-        - Consider ATS compatibility in all recommendations
-        - Keep responses concise and practical"""
+        system_prompt = """You are an expert resume and career consultant with deep knowledge of hiring practices, ATS systems, and industry requirements. You provide actionable, specific advice to help job seekers improve their resumes and career prospects. Always be encouraging but honest in your feedback."""
         
         if context:
-            system_prompt += f"\n\nResume Analysis Context:\n{context[:2000]}"
-    
-        
-        # Check if prompt is resume-related
-        resume_keywords = ['resume', 'cv', 'job', 'career', 'skills', 'experience', 'education', 'work', 'interview', 'hiring', 'ats', 'application', 'improve', 'better', 'add', 'remove', 'section', 'format']
-        if not any(keyword in prompt.lower() for keyword in resume_keywords):
-            return "I'm specialized in resume analysis and career advice. Please ask me questions related to your resume, job applications, career development, or interview preparation."
+            system_prompt += f"\n\nContext about the user's resume:\n{context[:2000]}"  # Limit context to avoid token limits
         
         data = {
             "model": self.model,
@@ -119,6 +103,12 @@ class ClaudeChatbot:
         try:
             response = requests.post(self.base_url, headers=headers, json=data, timeout=30)
             
+            # Handle rate limiting
+            if response.status_code == 429:
+                st.warning("Rate limit hit. Waiting 10 seconds...")
+                time.sleep(10)
+                response = requests.post(self.base_url, headers=headers, json=data, timeout=30)
+            
             # Handle payment required error
             if response.status_code == 402:
                 return "Error: This model requires payment. Please switch to a free model or add credits to your OpenRouter account."
@@ -135,7 +125,7 @@ class ClaudeChatbot:
             if e.response.status_code == 402:
                 return "Error: Payment required. This model is not free. Please use a free model or add credits."
             elif e.response.status_code == 429:
-                return "Error: Too many requests. Please try again in a moment."
+                return "Error: Rate limit exceeded. Please try again later."
             else:
                 return f"HTTP Error: {e.response.status_code} - {str(e)}"
         except requests.exceptions.RequestException as e:
@@ -160,6 +150,7 @@ class ClaudeChatbot:
         self.model = model_name
         st.success(f"Switched to model: {model_name}")
 
+
 # Download NLTK data if not already present
 @st.cache_resource
 def download_nltk_data():
@@ -593,12 +584,12 @@ class ResumeAnalyzer:
 
 def main():
     st.set_page_config(
-        page_title="AI Resume Analyzer with Claude",
+        page_title="AI Resume Analyzer with Chatbot",
         page_icon="📄",
         layout="wide"
     )
     
-    st.title("🚀 AI-Powered Resume Analyzer with Claude Assistant")
+    st.title("🚀 AI-Powered Resume Analyzer with Chatbot Assistant")
     st.markdown("Upload your resume and get comprehensive analysis with AI-powered insights!")
     
     # Show dependency and API status
@@ -624,9 +615,9 @@ def main():
         
         with col4:
             if os.getenv('OPENROUTER_API_KEY'):
-                st.success("✅ Claude API Available")
+                st.success("✅ Chatbot API Available")
             else:
-                st.error("❌ Claude API Not Available")
+                st.error("❌ Chatbot API Not Available")
     
     # Initialize analyzer
     try: