init

app.py

@@ -114,26 +114,19 @@ elif page == "💡 Brainstorm Hub":
 elif page == "🤖 AI Assistant":
     st.title("🤖 Prospira AI Business Mentor")
 
-    class LLaMAAssistant:
-        def __init__(self, model_name="meta-llama/Llama-2-7b-chat-hf"):
-            try:
-                self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-                self.model = AutoModelForCausalLM.from_pretrained(
-                    model_name, torch_dtype=torch.float16, device_map="auto"
-                )
-            except Exception as e:
-                st.error(f"Model loading error: {e}")
-                self.model = None
-                self.tokenizer = None
-
-        def generate_response(self, prompt: str) -> str:
-            if not self.model or not self.tokenizer:
-                return "LLM not initialized."
-            input_ids = self.tokenizer(prompt, return_tensors="pt").input_ids.to(self.model.device)
-            output = self.model.generate(input_ids, max_length=300)
-            return self.tokenizer.decode(output[0], skip_special_tokens=True)
-
-    assistant = LLaMAAssistant()
+    def render_chat():
+        st.header("💬 Business AI Assistant")
+        
+        # Embed the Hugging Face Space iframe
+        st.write("""
+        <iframe
+            src="https://demoorganisation34-chatbot-for-prospira.hf.space"
+            frameborder="0"
+            width="850"
+            height="450"
+        ></iframe>
+        """, unsafe_allow_html=True)
+
     user_input = st.text_area("Ask Prospira AI for Business Advice:")
     if st.button("Get AI Insights"):
         response = assistant.generate_response(user_input)