feat: upload photos

app.py

@@ -4,6 +4,16 @@ import requests
 API_URL = 'https://pic-gai.up.railway.app'
 # API_URL = 'http://localhost:8000'
 
+def gallery(column, images):
+    groups = []
+    for i in range(0, len(images), column):
+        groups.append(images[i:i+column])
+
+    for group in groups:
+        cols = st.columns(column)
+        for i, image in enumerate(group):
+            cols[i].image(image)
+
 st.title('CollageAI')
 
 # Input field for user prompt
@@ -12,26 +22,17 @@ user_prompt = st.text_area(
     placeholder="For our anniversary, I want to write a card to my partner to celebrate our love and share all the things I adore about them."
 )
 
+uploaded_images = st.file_uploader("Choose photos", accept_multiple_files=True)
+if uploaded_images:
+    st.subheader('Uploaded photos:')
+    gallery(4, uploaded_images)
+
 # pick number of photos
-photos_count = st.number_input(
-    'How many photos would you like to use?',
-    min_value=0,
-    max_value=10,
-)
+photos_count = len(uploaded_images)
 
 # Submit buttons for templates and stickers
 generate_button = st.button('Generate')
 
-def gallery(column, images):
-    groups = []
-    for i in range(0, len(images), column):
-        groups.append(images[i:i+column])
-
-    for group in groups:
-        cols = st.columns(column)
-        for i, image in enumerate(group):
-            cols[i].image(image)
-
 if generate_button:
     if user_prompt:
         # Prepare the params with the user prompt