Fixed all bugs, working condition -> Making a demo of this

app.py

@@ -5,9 +5,8 @@ import pandas as pd
 import numpy as np
 from datetime import datetime, timedelta
 from typing import Dict, List, Any
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 import streamlit as st
+import streamlit.components.v1 as components
 
 # --- Data Processing Class ---
 class DataProcessor:
@@ -360,19 +359,7 @@ def render_brainstorm_page():
             st.info("No products yet. Create one to get started!")
 
 
-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, context: list = None) -> str:
         if not self.model or not self.tokenizer:
@@ -412,20 +399,19 @@ class LLaMAAssistant:
 
 def render_chat():
     st.header("💬AI Business Mentor")
-     st.title("🤖 Prospira AI Business Mentor")
-    st.write("""
+    st.title("🤖 Prospira AI Business Mentor")
+
+    iframe_code = """
     <iframe
         src="https://demoorganisation34-chatbot-for-prospira.hf.space"
-        frameborder="0"
-        width="850"
-        height="450"
-    ></iframe>
-    """, unsafe_allow_html=True)
+        width="100%" height="500" frameborder="0">
+    </iframe>
+    """
+    components.html(iframe_code, height=600)
 
 def render_home():
     st.title("🚀 Welcome to Prospira")
     st.subheader("📊 Data-Driven Solutions for Businesses and Creators")
-
     st.markdown("""
     **Prospira** empowers businesses and creators to enhance their content, products, and marketing strategies using AI-driven insights.
     
@@ -440,22 +426,17 @@ def render_home():
 def main():
     st.set_page_config(
         page_title="Prospira",
-        page_icon="📊",
-        layout="wide",
+        page_icon="🚀",
+        layout="centered",
         initial_sidebar_state="expanded"
     )
-    
-    with st.sidebar:
-        st.title("Prospira")
-        st.subheader("Data-Driven Solutions")
-        
-        page = st.radio(
-            "Navigation",
-            ["Home", "Dashboard", "Analytics", "Brainstorm", "Chat"]
-        )
+
+    # Create a selection box to choose between pages
+    page = st.sidebar.radio("Select a page", ["Home", "Dashboard", "Analytics", "Brainstorm", "Chat"])
+
     if page == "Home":
         render_home()
-    if page == "Dashboard":
+    elif page == "Dashboard":
         render_dashboard()
     elif page == "Analytics":
         render_analytics()