fix: image rotation

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+from PIL import Image, ImageOps
+
 import requests
 
 API_URL = 'https://pic-gai.up.railway.app'
@@ -24,8 +26,9 @@ user_prompt = st.text_area(
 
 uploaded_images = st.file_uploader("Choose photos", accept_multiple_files=True)
 if uploaded_images:
-    st.subheader('Uploaded photos:')
-    gallery(4, uploaded_images)
+    images = [Image.open(image) for image in uploaded_images]
+    images = [ImageOps.exif_transpose(image) for image in images]
+    gallery(4, images)
 
 # pick number of photos
 photos_count = len(uploaded_images)
@@ -44,7 +47,8 @@ if generate_button:
         # remove empty params
         params = {k: v for k, v in params.items() if v is not None}
 
-        print(params)
+        st.markdown("---")
+
         # Templates
         with st.container():
             # Define the FastAPI server URL for templates
@@ -60,8 +64,7 @@ if generate_button:
                     image_urls = [template.get('image_medium') for template in templates]
                     
                     if image_urls:
-                        st.markdown("---")
-                        st.subheader('Generated templates:')
+                        st.subheader('Generated templates')
                         gallery(4, image_urls[:8])
                     else:
                         st.warning('No images were generated. Please try again with a different prompt.')
@@ -82,12 +85,27 @@ if generate_button:
                     image_urls = [sticker.get('image_url') for sticker in stickers]
                     
                     if image_urls:
-                        st.markdown("---")
-                        st.subheader('Generated stickers:')
+                        st.subheader('Generated stickers')
                         gallery(4, image_urls[:8])
                     else:
                         st.warning('No images were generated. Please try again with a different prompt.')
                 else:
                     st.error(f"Error: {response.status_code}")
+        
+        # Keywords
+        with st.container():
+            # Define the FastAPI server URL for keywords
+            url = f"{API_URL}/api/analyze_prompt"
+
+            # Make a request to the FastAPI server
+            response = requests.get(url, params=params)
+
+            # Display the response in the appropriate output container
+            if response.status_code == 200:
+                st.subheader('Keywords based on prompt')
+                keywords = response.json().get('keywords', [])
+                st.write(keywords)
+            else:
+                st.error(f"Error: {response.status_code}")
     else:
         st.warning('Please enter a prompt before submitting.')