Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,108 @@
+import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from PIL import Image
+import requests
+from io import BytesIO
+
+# Load Moondream model (assuming local inference; for Spaces, ensure GPU access)
+model_id = "vikhyatk/moondream2"
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True, torch_dtype=torch.float16)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model.to("cuda" if torch.cuda.is_available() else "cpu")
+
+def analyze_room(image, prompt):
+    """Use Moondream to analyze the uploaded room image and provide design suggestions."""
+    if image is None:
+        return "Please upload an image of your room."
+    
+    # Process image with Moondream
+    enc_image = model.encode_image(image).to("cuda" if torch.cuda.is_available() else "cpu")
+    response = model.answer_question(enc_image, prompt, tokenizer)
+    return response
+
+def generate_color_palette(suggestion_text):
+    """Generate color palette suggestions based on Moondream's output."""
+    # Simple mock for color palette; in practice, integrate with a color model or API
+    palettes = [
+        ["#FF6B6B", "#4ECDC4", "#45B7D1", "#96CEB4", "#FFEAA7"],
+        ["#DDA0DD", "#98D8C8", "#F7DC6F", "#BB8FCE", "#85C1E9"],
+        ["#F8C471", "#82E0AA", "#F1948A", "#AED6F1", "#D5DBDB"]
+    ]
+    return random.choice(palettes)
+
+def furniture_recommendations(style):
+    """Provide furniture recommendations based on style."""
+    recommendations = {
+        "modern": ["Minimalist sofa", "Glass coffee table", "Geometric rug"],
+        "traditional": ["Wooden armchair", "Crystal chandelier", "Oriental vase"],
+        "minimalist": ["White shelves", "Black accent chair", "Plant pot"]
+    }
+    return recommendations.get(style.lower(), ["Generic chair", "Table", "Lamp"])
+
+# Onboarding components
+with gr.Blocks(title="AI Interior Design Assistant", theme=gr.themes.Soft()) as demo:
+    gr.HTML("""
+    <div style="text-align: center; padding: 20px;">
+        <h1>🏠 AI Interior Design Assistant</h1>
+        <p>Transform your space with AI-powered design suggestions using Moondream vision model.</p>
+        <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #007acc; text-decoration: none;">Built with anycoder</a>
+    </div>
+    """)
+    
+    with gr.Tabs() as tabs:
+        with gr.TabItem("Welcome", id="welcome"):
+            gr.Markdown("""
+            ## Welcome to Your AI Design Journey! 🎨
+            - **Step 1:** Upload a photo of your room.
+            - **Step 2:** Describe your vision or ask questions.
+            - **Step 3:** Get personalized design suggestions, color palettes, and furniture ideas.
+            - **Step 4:** Iterate and refine with follow-up prompts.
+            """)
+            gr.Image(value="https://via.placeholder.com/600x400?text=Upload+Your+Room+Photo", label="Example Room")
+        
+        with gr.TabItem("Design Studio", id="studio"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    image_input = gr.Image(label="Upload Room Photo", type="pil")
+                    prompt_input = gr.Textbox(label="Describe your vision (e.g., 'Make it cozy and modern')", lines=3, placeholder="Ask for design ideas, color schemes, or specific changes...")
+                    analyze_btn = gr.Button("Analyze & Suggest", variant="primary")
+                
+                with gr.Column(scale=2):
+                    output_text = gr.Textbox(label="AI Design Suggestions", lines=10, interactive=False)
+                    with gr.Row():
+                        palette_output = gr.ColorPicker(label="Suggested Colors", interactive=False)
+                        furniture_output = gr.Textbox(label="Furniture Recommendations", lines=5, interactive=False)
+            
+            def full_analysis(image, prompt):
+                if not image:
+                    return "Upload an image first!", "", ""
+                analysis = analyze_room(image, prompt)
+                palette = generate_color_palette(analysis)
+                style = "modern" if "modern" in prompt.lower() else "traditional"  # Simple heuristic
+                furniture = furniture_recommendations(style)
+                return analysis, ", ".join(palette), "\n".join(furniture)
+            
+            analyze_btn.click(full_analysis, [image_input, prompt_input], [output_text, palette_output, furniture_output])
+        
+        with gr.TabItem("Gallery", id="gallery"):
+            gr.Markdown("## Inspiration Gallery")
+            # Mock gallery; in practice, load from a dataset
+            gr.Gallery([
+                "https://via.placeholder.com/300x200?text=Modern+Living+Room",
+                "https://via.placeholder.com/300x200?text=Cozy+Bedroom",
+                "https://via.placeholder.com/300x200?text=Kitchen+Renovation"
+            ], columns=3, height=400)
+            
+            gr.Markdown("Upload your designs to share!")
+            gallery_upload = gr.File(file_count="multiple", file_types=["image"])
+    
+    # Footer
+    gr.HTML("""
+    <div style="text-align: center; padding: 20px; color: gray;">
+        <p>Powered by Moondream Vision Model | Built with Gradio</p>
+    </div>
+    """)
+
+if __name__ == "__main__":
+    demo.launch()