Create app.py

app.py

@@ -0,0 +1,56 @@
+import streamlit as st
+import requests
+import os
+from PIL import Image
+
+# Streamlit app title
+st.title("Background Remover App")
+
+# Input for API Key
+api_key = st.text_input("Enter Stability API Key", type="password")
+
+# File uploader for images
+uploaded_files = st.file_uploader("Upload Images", type=["png", "jpg", "jpeg", "webp"], accept_multiple_files=True)
+
+# Output format selector
+output_format = st.selectbox("Select Output Format", options=["png", "webp"], index=0)
+
+# Process button
+if st.button("Start Processing"):
+    if not api_key:
+        st.error("Please enter your API key.")
+    elif not uploaded_files:
+        st.error("Please upload at least one image.")
+    else:
+        progress_bar = st.progress(0)
+        results = []
+        for i, uploaded_file in enumerate(uploaded_files):
+            image = uploaded_file.read()
+            try:
+                response = requests.post(
+                    "https://api.stability.ai/v2beta/stable-image/edit/remove-background",
+                    headers={
+                        "authorization": f"Bearer {api_key}",
+                        "accept": "image/*"
+                    },
+                    files={"image": image},
+                    data={"output_format": output_format},
+                )
+                
+                if response.status_code == 200:
+                    output_file_path = f"{uploaded_file.name.split('.')[0]}_processed.{output_format}"
+                    with open(output_file_path, "wb") as f:
+                        f.write(response.content)
+                    
+                    results.append((uploaded_file.name, output_file_path))
+                    st.success(f"Processed: {uploaded_file.name}")
+                    st.image(Image.open(output_file_path), caption=f"Processed: {output_file_path}")
+                else:
+                    st.error(f"Failed to process {uploaded_file.name}: {response.json()}")
+            except Exception as e:
+                st.error(f"Error processing {uploaded_file.name}: {e}")
+            progress_bar.progress((i + 1) / len(uploaded_files))
+        
+        if results:
+            st.success("Processing completed!")
+