push

app.py

@@ -0,0 +1,185 @@
+import os
+import gradio as gr
+from dotenv import load_dotenv
+
+load_dotenv()
+
+style_transfer_function = gr.load(
+    "prgarg007/ai-style-transfer-dreamsofa",
+    src="spaces",
+    hf_token=os.getenv("HF_TOKEN")
+)
+
+def process_with_private_space(swatch_image, product_image, room_image, room_style):
+    """
+    Call the private space as a function and return the results
+    """
+    if style_transfer_function is None:
+        return None, None, "Error: Could not connect to the AI processing service. Please try again later."
+    
+    if not swatch_image or not product_image or not room_image:
+        return None, None, "Please upload all three images (swatch, product, and room)."
+    
+    try:
+        # Call the private space as a function using the API name we defined
+        styled_product, final_result, status_message = style_transfer_function(
+            swatch_image, 
+            product_image, 
+            room_image, 
+            room_style,
+            api_name="process_style_transfer"
+        )
+        
+        return styled_product, final_result, status_message
+        
+    except Exception as e:
+        return None, None, f"Error processing images: {str(e)}"
+
+def create_public_interface():
+    """Create the public interface that uses the private space as a function"""
+    
+    with gr.Blocks(
+        title="AI Style Transfer & Room Integration - Public Access",
+        theme=gr.themes.Soft()
+    ) as app:
+        
+        gr.Markdown("""
+        # 🚀 AI Powered Style Transfer & Room Integration
+        ## Public Access Portal
+        
+        Transform your furniture with AI-powered style transfer and room integration!
+        
+        **How it works:**
+        1. Upload a **swatch image** (fabric/pattern you want to apply)
+        2. Upload a **product image** (furniture piece to style)
+        3. Upload a **room image** (where you want to place the furniture)
+        4. Select your preferred **room style**
+        5. Let AI work its magic! ✨
+        """)
+        
+        # Check if the private space is loaded
+        if style_transfer_function is None:
+            gr.Markdown("""
+            ⚠️ **Service Currently Unavailable**
+            
+            The AI processing service is currently unavailable. This might be due to:
+            - Network connectivity issues
+            - Authentication problems
+            - Service maintenance
+            
+            Please try again later or contact support.
+            """)
+            return app
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### 📤 Upload Your Images")
+                
+                swatch_input = gr.Image(
+                    label="🎨 Swatch/Pattern Image",
+                    type="pil",
+                    height=200,
+                    info="Upload an image of the fabric, pattern, or style you want to apply"
+                )
+                
+                product_input = gr.Image(
+                    label="📦 Furniture/Product Image", 
+                    type="pil",
+                    height=200,
+                    info="Upload an image of the furniture piece you want to restyle"
+                )
+                
+                room_input = gr.Image(
+                    label="🏠 Room/Background Image",
+                    type="pil", 
+                    height=200,
+                    info="Upload an image of the room where you want to place the furniture"
+                )
+                
+                room_style = gr.Dropdown(
+                    choices=["modern", "vintage", "industrial", "scandinavian", "bohemian", "rustic"],
+                    value="modern",
+                    label="🎭 Room Style Theme",
+                    info="Choose the overall aesthetic style for the room integration"
+                )
+                
+                process_btn = gr.Button(
+                    "🤖 Transform with AI", 
+                    variant="primary", 
+                    size="lg",
+                    scale=1
+                )
+                
+                gr.Markdown("""
+                ### 💡 Tips for Best Results:
+                - Use high-quality, clear images
+                - Ensure good lighting in your photos
+                - Choose images with similar perspectives
+                - Swatch images work best when the pattern/texture is clearly visible
+                """)
+            
+            with gr.Column(scale=1):
+                gr.Markdown("### 🎯 AI Generated Results")
+                
+                styled_product_output = gr.Image(
+                    label="🎨 Step 1: Styled Product",
+                    height=300,
+                    info="Your furniture with the new style applied"
+                )
+                
+                final_output = gr.Image(
+                    label="🏆 Step 2: Room Integration", 
+                    height=300,
+                    info="Your styled furniture integrated into the room scene"
+                )
+                
+                status_output = gr.Textbox(
+                    label="🔄 Processing Status",
+                    value="Ready to transform your furniture with AI...",
+                    interactive=False,
+                    info="Current processing status and any messages"
+                )
+                
+                gr.Markdown("""
+                ### 🚀 Processing Steps:
+                1. **AI Analysis**: Examines your swatch and product images
+                2. **Style Transfer**: Applies the swatch style to your furniture  
+                3. **Room Integration**: Places the styled furniture in your room
+                4. **Final Rendering**: Creates photorealistic results
+                
+                *Processing typically takes 30-60 seconds*
+                """)
+        
+        # Connect the button to our function that calls the private space
+        process_btn.click(
+            fn=process_with_private_space,
+            inputs=[swatch_input, product_input, room_input, room_style],
+            outputs=[styled_product_output, final_output, status_output],
+            show_progress=True
+        )
+        
+        # Add some example information
+        gr.Markdown("""
+        ---
+        ### 🎨 Example Use Cases:
+        - **Interior Design**: Preview how different fabrics look on your furniture
+        - **E-commerce**: Show products in various styles and room settings  
+        - **Home Renovation**: Experiment with different design aesthetics
+        - **Furniture Customization**: Visualize custom upholstery options
+        
+        ### ⚡ Powered by:
+        - OpenAI DALL-E for image generation
+        - Advanced AI for style analysis and transfer
+        - Gradio for seamless user experience
+        
+        ### 🔒 Privacy & Security:
+        - Your images are processed securely
+        - No images are permanently stored
+        - Processing happens in real-time
+        """)
+    
+    return app
+
+if __name__ == "__main__":
+    demo = create_public_interface()
+    demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio==5.38.2
+openai==1.97.1
+pillow==10.4.0
+python-dotenv==1.0.1