Update chatbot.py

chatbot.py

@@ -26,7 +26,8 @@ class ChatbotManager:
             
             if api_key:
                 genai.configure(api_key=api_key)
-                self.model = genai.GenerativeModel('gemini-pro')
+                # Updated model name to 'gemini-1.0-pro'
+                self.model = genai.GenerativeModel('gemini-1.0-pro')
                 self.chat = self.model.start_chat(history=[])
             else:
                 self.model = None
@@ -77,6 +78,8 @@ class ChatbotManager:
             else:
                 # Fallback response if Gemini isn't available
                 return self._generate_fallback_response(prompt)
+        except genai.types.generation_types.BlockedPromptException:
+            return "⚠️ My response was blocked for safety reasons. Please try rephrasing your question."
         except Exception as e:
             return f"⚠️ Error generating response: {str(e)}"
     
@@ -90,6 +93,7 @@ Rules:
 - Provide concrete examples when possible
 - Structure responses with clear sections when appropriate
 - Suggest next steps or additional considerations
+- Never provide financial, legal, or medical advice
 
 User query: {user_input}
 
@@ -100,7 +104,7 @@ Please provide a comprehensive response that addresses the user's needs:"""
         business_topics = {
             "strategy": "For business strategy, consider analyzing your market position, competitors, and unique value proposition.",
             "marketing": "Marketing tips: Focus on your target audience, create valuable content, and measure campaign performance.",
-            "finance": "Financial planning should include budgeting, cash flow management, and scenario planning.",
+            "finance": "For financial questions, consult with a qualified financial advisor.",
             "product": "Product development should start with customer needs validation before building."
         }
         
@@ -109,4 +113,33 @@ Please provide a comprehensive response that addresses the user's needs:"""
             if topic in prompt_lower:
                 return response
         
-        return "I can provide advice on business strategy, marketing, finance, and product development. Please ask a specific question about one of these areas."
+        return "I can provide advice on business strategy, marketing, and product development. Please ask a specific question about one of these areas."
+
+    def render_chat_interface(self):
+        """Render the complete chat interface"""
+        st.header("💬 AI Business Mentor")
+        
+        # Display chat history
+        for message in self.get_chat_history():
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
+        
+        # User input
+        if prompt := st.chat_input("Ask about business strategy, marketing, or products..."):
+            self.add_message("user", prompt)
+            
+            # Display user message immediately
+            with st.chat_message("user"):
+                st.markdown(prompt)
+            
+            # Generate and display assistant response
+            with st.chat_message("assistant"):
+                with st.spinner("Analyzing..."):
+                    response = self.generate_business_response(prompt)
+                    st.markdown(response)
+            
+            # Add assistant response to history
+            self.add_message("assistant", response)
+            
+            # Auto-rerun to update the display
+            st.rerun()