Update app.py

app.py

@@ -1,75 +1,20 @@
 import streamlit as st
-import cv2
-import numpy as np
-import plotly.graph_objects as go
 from PIL import Image
+import numpy as np
 
-def preprocess_image(image):
-    gray = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
-    orb = cv2.ORB_create()
-    keypoints, descriptors = orb.detectAndCompute(gray, None)
-    return keypoints, descriptors
-
-def match_features(desc1, desc2):
-    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
-    matches = bf.match(desc1, desc2)
-    matches = sorted(matches, key=lambda x: x.distance)
-    return matches[:50]  # Keep top 50 matches
-
-def create_point_cloud(images):
-    if len(images) < 2:
-        st.error("Please upload at least two images.")
-        return None
-
-    keypoints_list = []
-    descriptors_list = []
-    for image in images:
-        kp, desc = preprocess_image(image)
-        keypoints_list.append(kp)
-        descriptors_list.append(desc)
-
-    points_3d = []
-    for i in range(len(images) - 1):
-        matches = match_features(descriptors_list[i], descriptors_list[i+1])
-        for match in matches:
-            pt1 = keypoints_list[i][match.queryIdx].pt
-            pt2 = keypoints_list[i+1][match.trainIdx].pt
-            x = (pt1[0] + pt2[0]) / 2
-            y = (pt1[1] + pt2[1]) / 2
-            z = len(matches) - match.distance
-            points_3d.append([x, y, z])
-
-    return np.array(points_3d)
-
-def visualize_point_cloud(points):
-    fig = go.Figure(data=[go.Scatter3d(
-        x=points[:, 0],
-        y=points[:, 1],
-        z=points[:, 2],
-        mode='markers',
-        marker=dict(
-            size=2,
-            color=points[:, 2],
-            colorscale='Viridis',
-            opacity=0.8
-        )
-    )])
-    fig.update_layout(margin=dict(l=0, r=0, b=0, t=0))
-    return fig
+def main():
+    st.title("Simple Image Uploader")
+    st.write("Upload an image to display it.")
 
-st.title("3D Point Cloud Generation from 2D Images")
-st.write("Upload multiple images of an object from different angles to generate a basic point cloud.")
+    uploaded_file = st.file_uploader("Choose an image", type=['png', 'jpg', 'jpeg'])
 
-uploaded_files = st.file_uploader("Choose images", accept_multiple_files=True, type=['png', 'jpg', 'jpeg'])
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        st.image(image, caption='Uploaded Image', use_column_width=True)
 
-if uploaded_files:
-    images = [Image.open(file) for file in uploaded_files]
+        # Display image information
+        st.write(f"Image size: {image.size}")
+        st.write(f"Image format: {image.format}")
 
-    if st.button("Generate Point Cloud"):
-        with st.spinner("Processing..."):
-            point_cloud = create_point_cloud(images)
-            if point_cloud is not None and len(point_cloud) > 0:
-                fig = visualize_point_cloud(point_cloud)
-                st.plotly_chart(fig, use_container_width=True)
-            else:
-                st.error("Failed to create point cloud. Please try with different images.")
+if __name__ == "__main__":
+    main()