Update app.py

app.py

@@ -6,11 +6,11 @@ import numpy as np  # For numerical calculations
 # App Configuration
 st.set_page_config(page_title="Construction Site AR/VR Viewer", layout="wide")
 
-# Sidebar - Navigation and Filters
+# Sidebar - Navigation and Upload Section
 st.sidebar.title("Navigation")
-view_option = st.sidebar.radio("Choose View:", ["Home", "3D Visualization", "Issue Annotation", "Progress Updates"])
-st.sidebar.title("Filters")
-selected_area = st.sidebar.selectbox("Select Construction Area:", ["All", "Mechanical Systems", "Electrical Systems", "Plumbing Systems"])
+view_option = st.sidebar.radio("Choose View:", ["Home", "Upload Data", "3D Visualization", "Progress Updates"])
+st.sidebar.title("Upload Data")
+data_upload = st.sidebar.file_uploader("Upload Construction Site Data:", type=["csv", "json", "3d", "txt"])
 
 # App UI
 st.title("Construction Site Visualization with AR/VR")
@@ -24,37 +24,62 @@ def load_sample_model():
     mesh.compute_vertex_normals()
     return mesh
 
+# Process Uploaded Data
+def process_uploaded_data(uploaded_file):
+    try:
+        if uploaded_file.name.endswith(".csv"):
+            data = pd.read_csv(uploaded_file)
+            st.write("Uploaded CSV Data:")
+            st.dataframe(data)
+        elif uploaded_file.name.endswith(".json"):
+            data = pd.read_json(uploaded_file)
+            st.write("Uploaded JSON Data:")
+            st.json(data)
+        elif uploaded_file.name.endswith(".3d") or uploaded_file.name.endswith(".obj"):
+            st.write("3D Model Uploaded. Displaying in Visualization Section.")
+            return uploaded_file
+        else:
+            st.error("Unsupported file format. Please upload CSV, JSON, or 3D model files.")
+    except Exception as e:
+        st.error(f"Error processing file: {e}")
+
 if view_option == "Home":
     st.image("https://via.placeholder.com/800x400.png?text=Construction+Site+Visualization", caption="Welcome Screen", use_column_width=True)
     st.markdown(
-        "### Key Features:\n- AR/VR tools for real-time visualization.\n- Interactive walkthroughs.\n- Issue annotation directly in 3D environment.\n- Integration with progress updates and sensor data."
+        "### Key Features:\n- Upload site data for AR/VR visualization.\n- Interactive walkthroughs.\n- Monitor progress updates and site analytics."
     )
 
+elif view_option == "Upload Data":
+    st.subheader("Upload Site Data")
+    if data_upload:
+        uploaded_model = process_uploaded_data(data_upload)
+
 elif view_option == "3D Visualization":
     st.subheader("3D Visualization of Construction Site")
-    try:
-        model = load_sample_model()
-        o3d.visualization.draw_geometries([model])
-        st.success("3D model loaded successfully!")
-    except Exception as e:
-        st.error(f"Failed to load 3D model: {e}")
-
-elif view_option == "Issue Annotation":
-    st.subheader("Annotate Issues in 3D Environment")
-    st.text_input("Enter Issue Description:", placeholder="E.g., Incorrect pipe alignment in section B2")
-    uploaded_file = st.file_uploader("Upload Screenshot or 3D Model:", type=["png", "jpg", "3d"])
-    if st.button("Submit Annotation"):
-        st.success("Annotation submitted successfully!")
+    if data_upload and uploaded_model:
+        try:
+            st.write("Displaying Uploaded 3D Model")
+            model = load_sample_model()  # Replace with logic to load the uploaded model
+            o3d.visualization.draw_geometries([model])
+        except Exception as e:
+            st.error(f"Failed to load 3D model: {e}")
+    else:
+        st.info("Please upload a 3D model file in the Upload Data section.")
 
 elif view_option == "Progress Updates":
     st.subheader("Construction Progress Updates")
-    progress_data = {
-        "Area": ["Mechanical", "Electrical", "Plumbing"],
-        "Progress (%)": [75, 60, 90]
-    }
-    progress_df = pd.DataFrame(progress_data)
-    st.table(progress_df)
-    st.bar_chart(progress_df.set_index("Area"))
+    if data_upload:
+        st.write("Analyzing Progress from Uploaded Data...")
+        # Placeholder for progress analysis logic
+        progress_data = {
+            "Area": ["Mechanical", "Electrical", "Plumbing"],
+            "Progress (%)": [75, 60, 90]
+        }
+        progress_df = pd.DataFrame(progress_data)
+        st.table(progress_df)
+        st.bar_chart(progress_df.set_index("Area"))
+    else:
+        st.info("Please upload site data to view progress updates.")
 
 # Footer
 st.markdown("---")