Update app.py

app.py

@@ -18,21 +18,36 @@ def process_frame(frame):
 
 def main():
     st.title("Smart Mirror AI")
+    
+    # Option to run camera or upload image
     run = st.checkbox("Run Camera")
-    cap = cv2.VideoCapture(0)
-    frame_placeholder = st.empty()
+    uploaded_image = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
+    
+    if run:
+        cap = cv2.VideoCapture(0)
+        frame_placeholder = st.empty()
+        
+        while run:
+            success, frame = cap.read()
+            if not success:
+                st.write("Failed to capture video")
+                break
+            else:
+                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                predictions = process_frame(frame)
+                frame_placeholder.image(frame, channels="RGB")
+        cap.release()
     
-    while run:
-        success, frame = cap.read()
-        if not success:
-            st.write("Failed to capture video")
-            break
-        else:
-            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            predictions = process_frame(frame)
-            frame_placeholder.image(frame, channels="RGB")
-    cap.release()
+    elif uploaded_image is not None:
+        # If an image is uploaded, process it
+        image = Image.open(uploaded_image)
+        image = np.array(image)
+        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+        predictions = process_frame(image)
+        st.image(image, channels="BGR")
+        st.write(f"Predictions: {predictions}")
     
 if __name__ == "__main__":
     main()
 
+