Update helper.py

helper.py

@@ -9,17 +9,25 @@ from openai import OpenAI
 import streamlit as st
 import streamlit.components.v1 as components
 
-# Function to render custom HTML file uploader
 def custom_file_uploader():
     uploader_html = """
     <input type="file" id="fileUpload" accept=".png,.jpg,.jpeg" />
     <script>
-        // Hide the 'Take photo' option (works on some browsers, not universally enforced)
-        document.getElementById("fileUpload").capture = false;
+        const fileInput = document.getElementById("fileUpload");
+        fileInput.addEventListener("change", function(event) {
+            if (fileInput.files.length > 0) {
+                const file = fileInput.files[0];
+                const reader = new FileReader();
+                reader.readAsDataURL(file);
+                reader.onload = function() {
+                    const result = reader.result;
+                    window.parent.postMessage({type: 'FILE_UPLOAD', file: result}, '*');
+                };
+            }
+        });
     </script>
     """
-    # Embedding custom HTML uploader
-    components.html(uploader_html, height=100)
+    st.markdown(uploader_html, unsafe_allow_html=True)
 
 def resize_image(image: Image.Image) -> Image.Image:
     new_height = int(image.height * 512 / image.width)