import streamlit as st
import streamlit.components.v1 as components
import random
import json
import os
# Initialize session state
if "entities" not in st.session_state:
st.session_state.entities = []
# Function to get random 3D model file
def get_random_3d_model():
model_dir = "models"
model_files = [f for f in os.listdir(model_dir) if f.endswith(('.glb', '.obj'))]
return os.path.join(model_dir, random.choice(model_files)) if model_files else None
# Function to add a new entity to the scene
def add_entity(entity_type):
model_path = get_random_3d_model()
if model_path:
new_entity = {
'type': entity_type,
'model': model_path,
'position': {
'x': random.uniform(-5, 5),
'y': 0,
'z': random.uniform(-5, 5)
},
'rotation': {
'x': 0,
'y': random.uniform(0, 360),
'z': 0
}
}
st.session_state.entities.append(new_entity)
# Function to generate A-Frame entities
def generate_aframe_entities():
entities_html = ""
for entity in st.session_state.entities:
entities_html += f'''
'''
return entities_html
# HTML template
html_template = '''
Chofko's Ecosystem Simulator
{entities}
'''
# Streamlit app
def main():
st.set_page_config(page_title="Chofko's Ecosystem Simulator", layout="wide")
# Sidebar
st.sidebar.title("Chofko's Ecosystem Controls")
if st.sidebar.button("🚲 Add SkyBike"):
add_entity("skybike")
if st.sidebar.button("❤️ Add Heart"):
add_entity("heart")
if st.sidebar.button("🗑️ Clear All Entities"):
st.session_state.entities = []
st.sidebar.subheader("Current Entities")
st.sidebar.json(st.session_state.entities)
# Main area - HTML5 canvas
skybike_model = "models/SkyBike.glb"
heart_model = "models/Heart.obj"
entities_html = generate_aframe_entities()
html_content = html_template.format(
skybike_model=skybike_model,
heart_model=heart_model,
entities=entities_html
)
components.html(html_content, height=600)
if __name__ == "__main__":
main()