initial commit

app.py

@@ -0,0 +1,129 @@
+import gradio as gr
+import requests
+import io
+from PIL import Image
+from catboost import CatBoostRegressor
+from web import HTMLCode, CSSCode, footCode
+import numpy as np
+import joblib
+
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
+headers = {"Authorization": "Bearer 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"}
+defaultprompt = "round table with blue color and smooth edges, places in the living room"
+
+main_model = CatBoostRegressor()
+main_model.load_model("model.cbm")
+
+pred_vol = joblib.load('model_predvol.pkl')
+pred_wei = joblib.load('model_predwei.pkl')
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+def stablefurniture(originalprompt, ftype, purpose, texture, length, width, height):
+    prompt = f"full {ftype} places in the living room {originalprompt} {texture} texture for {purpose}, having {int(length)} millimeter length, {int(width)} millimeter width, and {int(height)} millimeter height, hd quality, full furniture, hyperrealistic, highly detailed, sharp focus, cinematic lighting, for commercial website"
+    print(prompt)
+    image_bytes = query(
+        {
+            "inputs": prompt,
+        }
+    )
+    image = Image.open(io.BytesIO(image_bytes)).convert("RGBA")
+    # print(f"length: {length},width: {width},height: {height}")
+    vol_pred = pred_vol.predict([[length, width, height]])
+    wei_pred = pred_wei.predict([[length, width, height, vol_pred[0]]])
+    prediction = main_model.predict([length, width, height, vol_pred[0], wei_pred[0]])
+    rubles = "₽ " + str(np.round(prediction))
+    return image, rubles
+
+with gr.Blocks(theme=gr.themes.Soft(), css=CSSCode) as demo:
+    gr.HTML(HTMLCode)
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            with gr.Row():
+                originalprompt = gr.Textbox(label="Prompt",default=defaultprompt)
+            with gr.Row():
+                ftype = gr.Dropdown(
+                    [
+                        "Table",
+                        "Rack",
+                        "Closet",
+                        "Cabinet",
+                        "Roll-out stand",
+                        "Pedestal",
+                        "Screen",
+                        "Console",
+                        "Reception Desk",
+                        "Mezzanine",
+                        "Penalty",
+                        "Classical",
+                    ],
+                    label="Type",
+                    info="Which type of furniture are you looking for?",
+                )
+                purpose = gr.Dropdown(
+                    [
+                        "computer",
+                        "for clothes",
+                        "for documents",
+                        "for negotiations",
+                        "for office",
+                        "for office equipment",
+                        "for receptionists",
+                        "for magazine",
+                        "roll-out stand",
+                        "writing",
+                    ],
+                    label="Purpose",
+                    info="How may your furnityre help you?",
+                )
+                texture = gr.Dropdown(
+                    [
+                        "Beech",
+                        "Oak",
+                        "Kraft white",
+                        "Sonoma oak Light",
+                        "Craft Golden",
+                        "Wenge/Oak",
+                        "Nut",
+                        "Wine",
+                        "Grey",
+                        "Oak Cronberg",
+                        "Cherry",
+                    ],
+                    label="Texture",
+                    info="How would you like it to be?",
+                )
+            with gr.Row():
+                length = gr.Number(label="Length")
+                width = gr.Number(label="Width")
+                height = gr.Number(label="Height")
+            btn = gr.Button("Dream")
+            prediction = gr.Textbox(label="Estimated Cost")
+        with gr.Column(scale=2, min_width=600):
+            furniture = gr.Image().style(height=580)
+    btn.click(
+        stablefurniture,
+        inputs=[originalprompt, ftype, texture, purpose, length, width, height],
+        outputs=[furniture, prediction],
+    )
+    with gr.Row():
+        gr.HTML(footCode)
+
+demo.launch()
+
+
+# gr.Interface(fn=stablefurniture, inputs=[
+#     gr.Textbox(),
+#     gr.Dropdown(
+#             ["Table","Rack","Closet","Cabinet","Roll-out stand","Pedestal","Screen","Console","Reception Desk","Mezzanine","Penalty","Classical"], label="Type", info="Which type of furniture are you looking for?"
+#         ),
+#     gr.Dropdown(
+#             ["computer","for clothes","for documents","for negotiations","for office","for office equipment","for receptionists","for magazine","roll-out stand","writing"], label="Purpose", info="Let us know why are you looking for this furniture.|"
+#         ),
+#     "number",
+#     "number",
+#     "number"],
+#     outputs=["image","number"],
+#     theme=gr.themes.Soft()).launch()

model_predvol.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c54b2c7b64a86c4a91ef4cba284c7813b1517a293713806c7345340b15cf91a0
+size 644

model_predwei.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6bd2ed115f641f60504789feccf62dc208321c7925b8397bc4794d51b0a150b8
+size 660

requirements.txt

@@ -0,0 +1,6 @@
+gradio==3.37.0
+requests==2.31.0
+Pillow==10.0.0
+catboost==1.2
+numpy==1.22.3
+pickle5==0.0.10

web.py

@@ -0,0 +1,176 @@
+HTMLCode = """
+            <head>
+                <link href="https://fonts.cdnfonts.com/css/vegapunk" rel="stylesheet">
+                <link href="https://fonts.cdnfonts.com/css/dustismo" rel="stylesheet">
+
+                
+            </head>
+            <style>
+            @import url('https://fonts.cdnfonts.com/css/vegapunk');
+            @import url('https://fonts.cdnfonts.com/css/dustismo');
+            h1{
+                font-family:"Vegapunk", sans-serif;
+            }
+            p{
+               font-family: 'Dustismo', sans-serif;
+                                                                                
+            }
+            </style>
+            <div style="text-align: center; margin: 0 auto;">
+              <div
+                style="
+                  display: inline-flex;
+                  align-items: center;
+                  gap: 0.8rem;
+                  font-size: 1.75rem;
+                "
+              >
+              <img src="https://seeklogo.com/images/U/ural-federal-university-logo-C368E954BC-seeklogo.com.png" style="height: 50px;" />
+                <h1 style="font-size: 50px; font-weight: 1200; margin-bottom: 7px;margin-top:5px;">
+                  AlternoVation
+                </h1>
+              <img src="https://td-alterna.ru/design/alterna/images/logo_alt.svg" style="width: 50px; height: 50px; object-fit: cover; object-position: 0% 0" />
+              </div>
+              <p style="margin-bottom: 10px; font-size: 20px; line-height: 23px;">
+                "Take your office game up a notch - upgrade your workspace with ease & boost productivity on a budget!"
+              </p>
+            </div>
+            
+        """
+
+footCode = """
+<p align="center">
+<iframe title="Report Section" width="720" height="480" 
+                src="https://app.powerbi.com/view?r=eyJrIjoiMTdjZWQ3ZmYtYzJkMy00YWM0LWFlODktNzkwOGU2NGU5ZDIyIiwidCI6ImYyM2M2Y2Y4LTVjY2MtNDhmMy04NzgzLWY3NWM3YzlkNjVkZiIsImMiOjl9" 
+                frameborder="0"
+                align="center" 
+                allowFullScreen="true">
+            </iframe>
+            </p>
+"""
+
+CSSCode = """
+        .gradio-container {
+            font-family: 'IBM Plex Sans', sans-serif;
+        }
+        .gr-button {
+            color: white;
+            border-color: black;
+            background: black;
+        }
+        input[type='range'] {
+            accent-color: black;
+        }
+        .dark input[type='range'] {
+            accent-color: #dfdfdf;
+        }
+        .container {
+            max-width: 730px;
+            margin: auto;
+            padding-top: 1.5rem;
+        }
+        #gallery {
+            min-height: 22rem;
+            margin-bottom: 15px;
+            margin-left: auto;
+            margin-right: auto;
+            border-bottom-right-radius: .5rem !important;
+            border-bottom-left-radius: .5rem !important;
+        }
+        #gallery>div>.h-full {
+            min-height: 20rem;
+        }
+        .details:hover {
+            text-decoration: underline;
+        }
+        .gr-button {
+            white-space: nowrap;
+        }
+        .gr-button:focus {
+            border-color: rgb(147 197 253 / var(--tw-border-opacity));
+            outline: none;
+            box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000);
+            --tw-border-opacity: 1;
+            --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color);
+            --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(3px var(--tw-ring-offset-width)) var(--tw-ring-color);
+            --tw-ring-color: rgb(191 219 254 / var(--tw-ring-opacity));
+            --tw-ring-opacity: .5;
+        }
+        #advanced-btn {
+            font-size: .7rem !important;
+            line-height: 19px;
+            margin-top: 12px;
+            margin-bottom: 12px;
+            padding: 2px 8px;
+            border-radius: 14px !important;
+        }
+        #advanced-options {
+            display: none;
+            margin-bottom: 20px;
+        }
+        .footer {
+            margin-bottom: 45px;
+            margin-top: 35px;
+            text-align: center;
+            border-bottom: 1px solid #e5e5e5;
+        }
+        .footer>p {
+            font-size: .8rem;
+            display: inline-block;
+            padding: 0 10px;
+            transform: translateY(10px);
+            background: white;
+        }
+        .dark .footer {
+            border-color: #303030;
+        }
+        .dark .footer>p {
+            background: #0b0f19;
+        }
+        .acknowledgments h4{
+            margin: 1.25em 0 .25em 0;
+            font-weight: bold;
+            font-size: 115%;
+        }
+        .animate-spin {
+            animation: spin 1s linear infinite;
+        }
+        @keyframes spin {
+            from {
+                transform: rotate(0deg);
+            }
+            to {
+                transform: rotate(360deg);
+            }
+        }
+        #share-btn-container {
+            display: flex; padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; width: 13rem;
+            margin-top: 10px;
+            margin-left: auto;
+        }
+        #share-btn {
+            all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.25rem !important; padding-bottom: 0.25rem !important;right:0;
+        }
+        #share-btn * {
+            all: unset;
+        }
+        #share-btn-container div:nth-child(-n+2){
+            width: auto !important;
+            min-height: 0px !important;
+        }
+        #share-btn-container .wrap {
+            display: none !important;
+        }
+        
+        .gr-form{
+            flex: 1 1 50%; border-top-right-radius: 0; border-bottom-right-radius: 0;
+        }
+        #prompt-container{
+            gap: 0;
+        }
+        #prompt-text-input, #negative-prompt-text-input{padding: .45rem 0.625rem}
+        #component-16{border-top-width: 1px!important;margin-top: 1em}
+        .image_duplication{position: absolute; width: 100px; left: 50px}
+
+        footer {visibility: hidden}
+"""