Spaces:
Sleeping
Sleeping
File size: 3,967 Bytes
d672462 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | import gradio as gr
import numpy as np
from PIL import Image, ImageDraw, ImageFont
def overlay_plan(image, sockets, switches, room_type, price_socket, price_switch, cable_per_point_m, cable_price_per_m, customer_view):
img = image.convert("RGBA")
w, h = img.size
draw = ImageDraw.Draw(img)
# simple suggested positions (rule-based)
# sockets along lower wall line; switches near "door" area left side
socket_points = []
switch_points = []
# distribute sockets on bottom area
for i in range(sockets):
x = int((i + 1) * w / (sockets + 1))
y = int(h * 0.83)
socket_points.append((x, y))
# switches on left side mid height
for i in range(switches):
x = int(w * 0.12)
y = int(h * (0.45 + (i - (switches-1)/2) * 0.12))
switch_points.append((x, y))
# draw overlays
def draw_marker(xy, label, color):
r = max(10, int(min(w, h) * 0.015))
x, y = xy
draw.ellipse((x-r, y-r, x+r, y+r), outline=color, width=4)
draw.text((x+r+6, y-r-6), label, fill=color)
for idx, p in enumerate(socket_points, start=1):
draw_marker(p, f"SD{idx}", (0, 255, 0, 255)) # Steckdose
for idx, p in enumerate(switch_points, start=1):
draw_marker(p, f"SW{idx}", (255, 165, 0, 255)) # Schalter
# rough cable estimate
points_total = sockets + switches
cable_m = points_total * max(0.0, float(cable_per_point_m))
# costs
cost_sockets = sockets * float(price_socket)
cost_switches = switches * float(price_switch)
cost_cable = cable_m * float(cable_price_per_m)
total = cost_sockets + cost_switches + cost_cable
if customer_view:
summary = (
f"Raumtyp: {room_type}\n"
f"Vorschlag: {sockets} Steckdosen, {switches} Schalter\n"
f"Geschätzte Leitung: {cable_m:.1f} m\n"
f"(Kosten ausgeblendet – Kundenansicht)"
)
else:
summary = (
f"Raumtyp: {room_type}\n"
f"Steckdosen: {sockets} x {price_socket:.2f} € = {cost_sockets:.2f} €\n"
f"Schalter: {switches} x {price_switch:.2f} € = {cost_switches:.2f} €\n"
f"Leitung: {cable_m:.1f} m x {cable_price_per_m:.2f} € = {cost_cable:.2f} €\n"
f"-----------------------------\n"
f"Summe (grob): {total:.2f} €"
)
return img.convert("RGB"), summary
with gr.Blocks(title="Smartelektric – Raum-Planer (Prototyp)") as demo:
gr.Markdown("## Smartelektric – Raum-Planer (Prototyp)\nFoto hochladen → Vorschlag + Material/Kosten grob")
with gr.Row():
inp = gr.Image(type="pil", label="Raumfoto (Rohbau / Bestand)")
out = gr.Image(type="pil", label="Vorschau mit Markierungen")
room_type = gr.Dropdown(
["Wohnzimmer", "Schlafzimmer", "Küche", "Bad", "Flur", "Büro", "Kinderzimmer", "Sonstiges"],
value="Wohnzimmer",
label="Raumtyp"
)
with gr.Row():
sockets = gr.Slider(0, 20, value=6, step=1, label="Anzahl Steckdosen (Vorschlag)")
switches = gr.Slider(0, 10, value=2, step=1, label="Anzahl Schalter (Vorschlag)")
with gr.Row():
price_socket = gr.Number(value=50, label="Preis pro Steckdose (inkl. z.B. 5m Leitung) €")
price_switch = gr.Number(value=25, label="Preis pro Schalter €")
with gr.Row():
cable_per_point_m = gr.Number(value=5, label="Leitung pro Punkt (m) – grobe Annahme")
cable_price_per_m = gr.Number(value=1.8, label="Leitungspreis pro Meter €")
customer_view = gr.Checkbox(value=False, label="Kundenansicht (Kosten ausblenden)")
summary = gr.Textbox(label="Auswertung", lines=10)
btn = gr.Button("Generieren")
btn.click(
overlay_plan,
inputs=[inp, sockets, switches, room_type, price_socket, price_switch, cable_per_point_m, cable_price_per_m, customer_view],
outputs=[out, summary]
)
demo.launch()
|