rows

app.py

@@ -9,59 +9,63 @@ def load_model():
     return loaded_pipeline
 
 # 예측 함수 정의
-def predict(construction_type, underground_floors, aboveground_floors, total_area, 
-            land_area, building_area, landscape_area, parking_lot_count, order_year, 
-            order_month, construction_period, location, building_purpose):
-            
+def predict_cost(구조, 지하층수, 지상층수, 주차수,  연면적, 대지면적, 건축면적, 조경면적, 발주년, 발주월, 공사기간, 현장위치, 건축법상용도):
     new_project = {
-        '구조': [construction_type],
-        '지하층수': [underground_floors],
-        '지상층수': [aboveground_floors],
-        '연면적': [total_area],
-        '대지면적': [land_area],
-        '건축면적': [building_area],
-        '조경면적': [landscape_area],
-        '주차수': [parking_lot_count],
-        '발주년': [order_year],  
-        '발주월': [order_month],  
-        '공사기간': [construction_period],
-        '현장위치': [location],
-        '건축법상용도': [building_purpose],
+        '구조': [구조],
+        '지하층수': [지하층수],
+        '지상층수': [지상층수],
+        '주차수': [주차수],
+        '연면적': [연면적],
+        '대지면적': [대지면적],
+        '건축면적': [건축면적],
+        '조경면적': [조경면적],
+        '발주년': [발주년],
+        '발주월': [발주월],
+        '공사기간': [공사기간],
+        '현장위치': [현장위치],
+        '건축법상용도': [건축법상용도],
     }
-    
+       
     model = load_model()
-    prediction = model.predict(pd.DataFrame.from_dict(new_project))
-    return prediction[0].tolist()
+    pred = model.predict(pd.DataFrame.from_dict(new_project))
+    return [f'{cost:,.0f}원' for cost in pred[0]]
 
-# Gradio 인터페이스 생성
-iface = gr.Interface(
-    fn=predict,
-    inputs=[
-        gr.Textbox(label="구조", default="철근콘크리트조"),
-        gr.Number(label="지하층수", default=1),
-        gr.Number(label="지상층수", default=1),
-        gr.Number(label="연면적", default=300),
-        gr.Number(label="대지면적", default=500),
-        gr.Number(label="건축면적", default=120),
-        gr.Number(label="조경면적", default=150),
-        gr.Number(label="주차수", default=50),
-        gr.Number(label="발주년", default=2023),
-        gr.Number(label="발주월", default=9),
-        gr.Number(label="공사기간", default=100),
-        gr.Textbox(label="현장위치", default="경기도"),
-        gr.Textbox(label="건축법상용도", default="업무시설"),
-    ],
-    outputs=[
-        gr.Textbox(label="건축공사비"),
-        gr.Textbox(label="기계공사비"),
-        gr.Textbox(label="전기공사비"),
-        gr.Textbox(label="통신공사비"),
-        gr.Textbox(label="토목공사비"),
-        gr.Textbox(label="조경공사비")
-    ],
-    title="HG 공사비 예측 서비스",
-    description="프로젝트 개요를 기반으로 공사비를 예측합니다."
-)
+with gr.Blocks() as demo:
+    gr.Markdown("HG 공사비 예측 모델")
+    with gr.Row():
+        구조 = gr.Dropdown(label="구조", choices=['철근콘크리트조', ''], value='철근콘크리트조')  # '...'부분에 다른 구조 타입 추가
+    with gr.Row():
+        지하층수 = gr.Number(label="지하층수", value=1)
+        지상층수 = gr.Number(label="지상층수", value=1)
+        주차수 = gr.Number(label="주차대수", value=50)
+    with gr.Row():
+        연면적 = gr.Number(label="연면적(m2)", value=300)
+        대지면적 = gr.Number(label="대지면적(m2)", value=500)
+        건축면적 = gr.Number(label="건축면적(m2)", value=120)
+        조경면적 = gr.Number(label="조경면적(m2)", value=150)
+    with gr.Row():
+        발주년 = gr.Number(label="발주년", value=2023)
+        발주월 = gr.Number(label="발주월", value=9)
+        공사기간 = gr.Number(label="공사기간(일)", value=100)
+    with gr.Row():
+        현장위치 = gr.Dropdown(label="현장위치", choices=['경기도', ''], value='경기도')  # '...'부분에 다른 현장 위치 추가
+        건축법상용도 = gr.Dropdown(label="건축법상용도", choices=['업무시설', ''], value='업무시설')  # '...'부분에 다른 건축법상용도 추가
+    
+    btn = gr.Button("예측")
+    
+    btn.click(
+        predict_cost,
+        inputs=[
+            구조, 지하층수, 지상층수, 주차수, 연면적, 대지면적, 건축면적, 조경면적,  발주년, 발주월, 공사기간, 현장위치, 건축법상용도
+        ],
+        outputs=[
+            gr.Textbox(label="건축공사비"),
+            gr.Textbox(label="기계공사비"),
+            gr.Textbox(label="전기공사비"),
+            gr.Textbox(label="통신공사비"),
+            gr.Textbox(label="토목공사비"),
+            gr.Textbox(label="조경공사비")
+        ]
+    )
 
-# Gradio 앱을 실행합니다.
-iface.launch()
+demo.launch()