Update app.py

app.py

@@ -21,14 +21,35 @@ def load_clients():
 
 caption_client, story_client = load_clients()
 
-# —––––––– Main Flow
-uploaded = st.file_uploader("Upload a child-friendly image:", type=["jpg", "png", "jpeg"])
-if not uploaded:
-    st.info("🌈 Please upload an image to start the magic!")
-else:
-    # Process Image
-    img = Image.open(uploaded).convert("RGB")
+# —––––––– Modified image processing (safe version) —–––––––
+def process_image(uploaded_file):
+    try:
+        img = Image.open(uploaded_file).convert("RGB")
+        
+        # Only resize if absolutely necessary
+        if img.size[0] > 2048 or img.size[1] > 2048:
+            img.thumbnail((2048, 2048))
+            
+        return img
+    except Exception as e:
+        st.error(f"Image processing error: {str(e)}")
+        st.stop()
+# —––––––– Main flow —–––––––
+uploaded = st.file_uploader("Upload an image:", type=["jpg", "png", "jpeg"])
+if uploaded:
+    img = process_image(uploaded)
     st.image(img, use_column_width=True)
+    
+    with st.spinner("Generating caption..."):
+        caption = generate_caption(img)
+        
+        # Your original validation
+        if not caption:
+            st.error("😢 Couldn't understand this image. Try another one!")
+            st.stop()
+            
+    st.success(f"**Caption:** {caption}")
+    
 
     # Generate Caption
     with st.spinner("🔍 Discovering image secrets..."):