Update app.py

app.py

@@ -31,7 +31,7 @@ from urllib.parse import quote
 from xml.etree import ElementTree as ET
 from openai import OpenAI
 
-# 1. Configuration and Setup
+# 1. 🚲BikeAI🏆 Configuration and Setup
 Site_Name = '🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI'
 title = "🚲BikeAI🏆 Claude and GPT Multi-Agent Research AI"
 helpURL = 'https://huggingface.co/awacke1'
@@ -50,7 +50,7 @@ st.set_page_config(
     }
 )
 
-# 2. Load environment variables and initialize clients
+# 2. 🚲BikeAI🏆 Load environment variables and initialize clients
 load_dotenv()
 
 # OpenAI setup
@@ -63,13 +63,13 @@ openai_client = OpenAI(
     organization=os.getenv('OPENAI_ORG_ID')
 )
 
-# 3. Claude setup
+# 3.🚲BikeAI🏆 Claude setup
 anthropic_key = os.getenv("ANTHROPIC_API_KEY_3")
 if anthropic_key == None:
     anthropic_key = st.secrets["ANTHROPIC_API_KEY"]
 claude_client = anthropic.Anthropic(api_key=anthropic_key)
 
-# 4. Initialize session states
+# 4.🚲BikeAI🏆 Initialize session states
 if 'transcript_history' not in st.session_state:
     st.session_state.transcript_history = []
 if "chat_history" not in st.session_state:
@@ -81,26 +81,17 @@ if "messages" not in st.session_state:
 if 'last_voice_input' not in st.session_state:
     st.session_state.last_voice_input = ""
 
-# 5. # HuggingFace setup
+# 5. 🚲BikeAI🏆 HuggingFace AI setup
 API_URL = os.getenv('API_URL')
 HF_KEY = os.getenv('HF_KEY')
 MODEL1 = "meta-llama/Llama-2-7b-chat-hf"
 MODEL2 = "openai/whisper-small.en"
-
 headers = {
     "Authorization": f"Bearer {HF_KEY}",
     "Content-Type": "application/json"
 }
 
-# Initialize session states
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-if "openai_model" not in st.session_state:
-    st.session_state["openai_model"] = "gpt-4o-2024-05-13"
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Custom CSS
+# 🚲BikeAI🏆 Custom CSS
 st.markdown("""
 <style>
     .main {
@@ -141,59 +132,6 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-# Bike Collections
-bike_collections = {
-    "Celestial Collection 🌌": {
-        "Eclipse Vaulter": {
-            "prompt": """Cinematic shot of a sleek black mountain bike silhouetted against a total solar eclipse. 
-                     The corona creates an ethereal halo effect, with lens flares accentuating key points of the frame.
-                     Dynamic composition shows the bike mid-leap, with stardust particles trailing behind.
-                     Camera angle: Low angle, wide shot
-                     Lighting: Dramatic rim lighting from eclipse
-                     Color palette: Deep purples, cosmic blues, corona gold""",
-            "emoji": "🌑"
-        },
-        "Starlight Leaper": {
-            "prompt": """A black bike performing an epic leap under a vast Milky Way galaxy.
-                     Shimmering stars blanket the sky while the bike's wheels leave a trail of stardust.
-                     Camera angle: Wide-angle upward shot
-                     Lighting: Natural starlight with subtle rim lighting
-                     Color palette: Deep blues, silver highlights, cosmic purples""",
-            "emoji": "✨"
-        },
-        "Moonlit Hopper": {
-            "prompt": """A sleek black bike mid-hop over a moonlit meadow,
-                     the full moon illuminating the misty surroundings. Fireflies dance around the bike,
-                     and soft shadows create a serene yet dynamic atmosphere.
-                     Camera angle: Side profile with slight low angle
-                     Lighting: Soft moonlight with atmospheric fog
-                     Color palette: Silver blues, soft whites, deep shadows""",
-            "emoji": "🌙"
-        }
-    },
-    "Nature-Inspired Collection 🌲": {
-        "Shadow Grasshopper": {
-            "prompt": """A black bike jumping between forest paths,
-                     with dappled sunlight streaming through the canopy. Shadows dance on the bike's frame
-                     as it soars above mossy logs.
-                     Camera angle: Through-the-trees tracking shot
-                     Lighting: Natural forest lighting with sun rays
-                     Color palette: Forest greens, golden sunlight, deep shadows""",
-            "emoji": "🦗"
-        },
-        "Onyx Leapfrog": {
-            "prompt": """A bike with obsidian-black finish jumping over a sparkling creek,
-                     the reflection on the water broken into ripples by the leap. The surrounding forest
-                     is vibrant with greens and browns.
-                     Camera angle: Low angle from water level
-                     Lighting: Golden hour side lighting
-                     Color palette: Deep blacks, water blues, forest greens""",
-            "emoji": "🐸"
-        }
-    }
-}
-
-
 # Helper Functions
 def generate_filename(prompt, file_type):
     """Generate a safe filename using the prompt and file type."""
@@ -565,7 +503,7 @@ def create_media_gallery():
     """Create the media gallery interface."""
     st.header("🎬 Media Gallery")
     
-    tabs = st.tabs(["🖼️ Images", "🎵 Audio", "🎥 Video", "🎨 Scene Generator"])
+    tabs = st.tabs(["🖼️ Images", "🎵 Audio", "🎥 Video"])
     
     with tabs[0]:
         image_files = glob.glob("*.png") + glob.glob("*.jpg")
@@ -603,25 +541,7 @@ def create_media_gallery():
                                                     "Describe what's happening in this video.")
                     st.markdown(analysis)
     
-    with tabs[3]:
-        for collection_name, bikes in bike_collections.items():
-            st.subheader(collection_name)
-            cols = st.columns(len(bikes))
-            
-            for idx, (bike_name, details) in enumerate(bikes.items()):
-                with cols[idx]:
-                    st.markdown(f"""
-                    <div class='bike-card'>
-                        <h3>{details['emoji']} {bike_name}</h3>
-                        <p>{details['prompt']}</p>
-                    </div>
-                    """, unsafe_allow_html=True)
-                    
-                    if st.button(f"Generate {bike_name} Scene"):
-                        prompt = details['prompt']
-                        # Here you could integrate with image generation API
-                        st.write(f"Generated scene description for {bike_name}:")
-                        st.write(prompt)
+
 
 def display_file_manager():
     """Display file management sidebar with guaranteed unique button keys."""
@@ -973,7 +893,7 @@ def create_media_gallery():
     """Create the media gallery interface."""
     st.header("🎬 Media Gallery")
     
-    tabs = st.tabs(["🖼️ Images", "🎵 Audio", "🎥 Video", "🎨 Scene Generator"])
+    tabs = st.tabs(["🖼️ Images", "🎵 Audio", "🎥 Video"])
     
     with tabs[0]:
         image_files = glob.glob("*.png") + glob.glob("*.jpg")
@@ -1011,25 +931,7 @@ def create_media_gallery():
                                                     "Describe what's happening in this video.")
                     st.markdown(analysis)
     
-    with tabs[3]:
-        for collection_name, bikes in bike_collections.items():
-            st.subheader(collection_name)
-            cols = st.columns(len(bikes))
-            
-            for idx, (bike_name, details) in enumerate(bikes.items()):
-                with cols[idx]:
-                    st.markdown(f"""
-                    <div class='bike-card'>
-                        <h3>{details['emoji']} {bike_name}</h3>
-                        <p>{details['prompt']}</p>
-                    </div>
-                    """, unsafe_allow_html=True)
-                    
-                    if st.button(f"Generate {bike_name} Scene"):
-                        prompt = details['prompt']
-                        # Here you could integrate with image generation API
-                        st.write(f"Generated scene description for {bike_name}:")
-                        st.write(prompt)
+
 
 def get_media_html(media_path, media_type="video", width="100%"):
     """Generate HTML for media player."""