Update app.py

app.py

@@ -13,7 +13,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS for better styling
+# Custom CSS for better styling with Renesistech branding
 st.markdown("""
 <style>
     .stApp {
@@ -34,9 +34,31 @@ st.markdown("""
         margin: 20px 0;
         background-color: #f8f9fa;
     }
+    .renesistech-header {
+        background-color: #ffffff;
+        padding: 20px;
+        margin-bottom: 30px;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        text-align: center;
+    }
+    .renesistech-footer {
+        background-color: #f8f9fa;
+        padding: 20px;
+        text-align: center;
+        border-top: 1px solid #e0e0e0;
+        margin-top: 40px;
+    }
 </style>
 """, unsafe_allow_html=True)
 
+# Add Renesistech header
+st.markdown("""
+<div class='renesistech-header'>
+    <h1 style='color: #333; margin: 0;'>Renesistech</h1>
+    <p style='color: #666; margin: 10px 0 0 0;'>Virtual Try-On Solution</p>
+</div>
+""", unsafe_allow_html=True)
+
 # Title and description
 st.title("✨ Virtual Try-On Application")
 st.markdown("""Upload a garment image and a person's photo to see how the garment would look on them!""")
@@ -47,58 +69,32 @@ col1, col2 = st.columns(2)
 # Garment image upload
 with col1:
     st.subheader("Garment Image")
-    garment_source = st.radio(
-        "Choose garment image source:",
-        ["Upload", "URL"],
-        key="garment_source"
-    )
-    
-    if garment_source == "Upload":
-        garment_file = st.file_uploader("Upload garment image", type=["jpg", "jpeg", "png", "webp"], key="garment_upload")
-        if garment_file:
-            st.image(garment_file, caption="Uploaded Garment", use_container_width=True)
-            # Save uploaded file to temporary URL
-            garment_img = Image.open(garment_file)
-            garment_buffer = BytesIO()
-            garment_img.save(garment_buffer, format="PNG")
-            garment_url = "temp_garment.png"
-            with open(garment_url, "wb") as f:
-                f.write(garment_buffer.getvalue())
-    else:
-        garment_url = st.text_input("Enter garment image URL", key="garment_url")
-        if garment_url:
-            try:
-                st.image(garment_url, caption="Garment from URL", use_container_width=True)
-            except:
-                st.error("Unable to load image from URL")
+    garment_file = st.file_uploader("Upload garment image", type=["jpg", "jpeg", "png", "webp"], key="garment_upload")
+    garment_url = None
+    if garment_file:
+        st.image(garment_file, caption="Uploaded Garment", use_container_width=True)
+        # Save uploaded file to temporary file
+        garment_img = Image.open(garment_file)
+        garment_buffer = BytesIO()
+        garment_img.save(garment_buffer, format="PNG")
+        garment_url = "temp_garment.png"
+        with open(garment_url, "wb") as f:
+            f.write(garment_buffer.getvalue())
 
 # Person image upload
 with col2:
     st.subheader("Person Image")
-    person_source = st.radio(
-        "Choose person image source:",
-        ["Upload", "URL"],
-        key="person_source"
-    )
-    
-    if person_source == "Upload":
-        person_file = st.file_uploader("Upload person image", type=["jpg", "jpeg", "png", "webp"], key="person_upload")
-        if person_file:
-            st.image(person_file, caption="Uploaded Person", use_container_width=True)
-            # Save uploaded file to temporary URL
-            person_img = Image.open(person_file)
-            person_buffer = BytesIO()
-            person_img.save(person_buffer, format="PNG")
-            person_url = "temp_person.png"
-            with open(person_url, "wb") as f:
-                f.write(person_buffer.getvalue())
-    else:
-        person_url = st.text_input("Enter person image URL", key="person_url")
-        if person_url:
-            try:
-                st.image(person_url, caption="Person from URL", use_container_width=True)
-            except:
-                st.error("Unable to load image from URL")
+    person_file = st.file_uploader("Upload person image", type=["jpg", "jpeg", "png", "webp"], key="person_upload")
+    person_url = None
+    if person_file:
+        st.image(person_file, caption="Uploaded Person", use_container_width=True)
+        # Save uploaded file to temporary file
+        person_img = Image.open(person_file)
+        person_buffer = BytesIO()
+        person_img.save(person_buffer, format="PNG")
+        person_url = "temp_person.png"
+        with open(person_url, "wb") as f:
+            f.write(person_buffer.getvalue())
 
 # Garment description input
 st.subheader("Garment Description")
@@ -139,10 +135,10 @@ if st.button("Generate Try-On", type="primary"):
         except Exception as e:
             st.error(f"Error occurred: {str(e)}")
 
-# Footer
-st.markdown("---")
+# Renesistech Footer
 st.markdown("""
-<div style='text-align: center; color: #666;'>
-    <p>Powered by IDM-VTON model via Replicate API</p>
+<div class='renesistech-footer'>
+    <p style='color: #333; font-weight: bold; margin: 0;'>© 2024 Renesistech</p>
+    <p style='color: #666; margin: 5px 0 0 0;'>Innovating Fashion Technology</p>
 </div>
 """, unsafe_allow_html=True)