Update app.py

app.py

@@ -17,6 +17,7 @@ from datetime import datetime
 from PyPDF2 import PdfReader
 import threading
 from PIL import Image
+from streamlit_javascript import st_javascript
 
 # ==============================================================================
 # Configuration & Constants
@@ -148,19 +149,26 @@ def update_player_state(username, position=None):
         players[username]["last_action_timestamp"] = time.time()
     state["players"] = players
     write_history_file(state)
+    return state
 
-def add_object_to_state(obj_data):
+def add_object_to_state(obj_data, username):
     state = read_history_file()
     state = prune_inactive_players(state)
     state["objects"][obj_data["obj_id"]] = obj_data
     write_history_file(state)
+    log_action(username, "place", obj_data)
+    return state
 
-def remove_object_from_state(obj_id):
+def remove_object_from_state(obj_id, username):
     state = read_history_file()
     state = prune_inactive_players(state)
     if obj_id in state["objects"]:
+        obj_data = state["objects"][obj_id]
         del state["objects"][obj_id]
         write_history_file(state)
+        log_action(username, "delete", {"obj_id": obj_id})
+        return state, obj_data
+    return state, None
 
 def log_action(username, action_type, data):
     timestamp = get_current_time_str()
@@ -184,6 +192,41 @@ def log_action(username, action_type, data):
     except Exception as e:
         print(f"Error writing to player history log {player_log_file}: {e}")
 
+# ==============================================================================
+# JavaScript Message Handling
+# ==============================================================================
+
+def handle_js_messages():
+    message = st_javascript("""
+        window.addEventListener('message', (event) => {
+            return JSON.stringify(event.data);
+        }, {once: true});
+        return null;
+    """)
+    if message:
+        try:
+            data = json.loads(message)
+            action = data.get("type")
+            payload = data.get("payload", {})
+            username = payload.get("username", st.session_state.username)
+            if action == "place_object":
+                state = add_object_to_state(payload["object_data"], username)
+                st.session_state.world_state = state
+                st.rerun()
+            elif action == "delete_object":
+                state, obj_data = remove_object_from_state(payload["obj_id"], username)
+                st.session_state.world_state = state
+                st.rerun()
+            elif action == "move_player":
+                state = update_player_state(username, payload["position"])
+                log_action(username, "move", payload["position"])
+                st.session_state.world_state = state
+                st.rerun()
+        except json.JSONDecodeError:
+            print(f"Invalid JS message: {message}")
+        except Exception as e:
+            print(f"Error handling JS message: {e}")
+
 # ==============================================================================
 # User State & Session Init
 # ==============================================================================
@@ -206,7 +249,7 @@ def load_username():
 
 def init_session_state():
     defaults = {
-        'message_input': "",
+        'message_counter': 0,
         'audio_cache': {},
         'tts_voice': "en-US-AriaNeural",
         'chat_history': [],
@@ -217,7 +260,8 @@ def init_session_state():
         'last_message': "",
         'selected_object': 'None',
         'paste_image_base64': "",
-        'new_world_name': "MyWorld"
+        'new_world_name': "MyWorld",
+        'world_state': {"objects": {}, "players": {}}
     }
     for k, v in defaults.items():
         if k not in st.session_state:
@@ -332,6 +376,7 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
     if st.session_state.get('enable_audio', True):
         tts_message = message
         audio_file = await async_edge_tts_generate(tts_message, voice, username)
+    log_action(username, "chat", {"message": message})
     return md_file, audio_file
 
 async def load_chat_history():
@@ -613,12 +658,14 @@ def render_sidebar():
                 if st.session_state.selected_object != name:
                     st.session_state.selected_object = name
                     update_player_state(st.session_state.username)
+                    st.rerun()
             col_idx += 1
         st.markdown("---")
         if st.button("🚫 Clear Tool", key="clear_tool", use_container_width=True):
             if st.session_state.selected_object != 'None':
                 st.session_state.selected_object = 'None'
                 update_player_state(st.session_state.username)
+                st.rerun()
 
         st.markdown("---")
         st.header("🗣️ Voice & User")
@@ -650,8 +697,9 @@ def render_main_content():
 
     with tab_world:
         st.header("Shared 3D World")
-        st.caption("Click to place objects with the selected tool. Right-click to delete. State is saved in history.json.")
+        st.caption("Click to place objects with the selected tool, or click to move player. Right-click to delete. State is saved in history.json.")
         state = read_history_file()
+        st.session_state.world_state = state
         html_file_path = 'index.html'
         try:
             with open(html_file_path, 'r', encoding='utf-8') as f:
@@ -666,6 +714,7 @@ def render_main_content():
 </script>"""
             html_content_with_state = html_template.replace('</head>', js_injection_script + '\n</head>', 1)
             components.html(html_content_with_state, height=700, scrolling=False)
+            handle_js_messages()
         except FileNotFoundError:
             st.error(f"CRITICAL ERROR: Could not find '{html_file_path}'.")
         except Exception as e:
@@ -682,7 +731,7 @@ def render_main_content():
             else:
                 st.caption("No chat messages yet.")
 
-        message_value = st.text_input("Your Message:", key="message_input", label_visibility="collapsed")
+        message_value = st.text_input("Your Message:", key=f"message_input_{st.session_state.get('message_counter', 0)}", label_visibility="collapsed")
         send_button_clicked = st.button("Send Chat 💬", key="send_chat_button")
         should_autosend = st.session_state.get('autosend', False) and message_value
 
@@ -693,7 +742,7 @@ def render_main_content():
                 voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
                 asyncio.run(save_chat_entry(st.session_state.username, message_to_send, voice))
                 update_player_state(st.session_state.username)
-                st.session_state.message_input = ""
+                st.session_state.message_counter = st.session_state.get('message_counter', 0) + 1
                 st.rerun()
             elif send_button_clicked:
                 st.toast("Message empty or same as last.")
@@ -705,6 +754,7 @@ def render_main_content():
         if st.button("Clear World State", key="clear_world_state"):
             state = {"objects": {}, "players": {}}
             write_history_file(state)
+            st.session_state.world_state = state
             st.success("World state cleared!")
             st.rerun()