Update app.py

app.py

@@ -343,6 +343,24 @@ def load_examples_from_directory(base_dir):
 
 def create_gradio_interface():
     predictor = ClimatePredictor('best_model.pth')
+
+    def process_elevation_file(file_obj):
+        if isinstance(file_obj, str):  # 파일 경로인 경우
+            return np.load(file_obj)
+        else:  # UploadedFile 객체인 경우
+            return np.load(file_obj.name)
+
+    def predict_with_processing(*args):
+        rgb_image, ndvi_image, terrain_image, elevation_file = args[:4]
+        weather_params = args[4:]
+        
+        # elevation 파일 처리
+        elevation_data = process_elevation_file(elevation_file)
+        
+        return predictor.predict_from_inputs(
+            rgb_image, ndvi_image, terrain_image, elevation_data,
+            *weather_params
+        )
     
     with gr.Blocks() as interface:
         gr.Markdown("# Renewable Energy Potential Predictor")
@@ -351,10 +369,10 @@ def create_gradio_interface():
         with gr.Row():
             # 입력 섹션 (1/3 크기)
             with gr.Column(scale=1):
-                rgb_input = gr.Image(label="RGB Satellite Image", type="numpy")
-                ndvi_input = gr.Image(label="NDVI Image", type="numpy")
-                terrain_input = gr.Image(label="Terrain Map", type="numpy")
-                elevation_input = gr.File(label="Elevation Data (NPY file)")
+                rgb_input = gr.Image(label="RGB Satellite Image", type="numpy", height=200)
+                ndvi_input = gr.Image(label="NDVI Image", type="numpy", height=200)
+                terrain_input = gr.Image(label="Terrain Map", type="numpy", height=200)
+                elevation_input = gr.File(label="Elevation Data (NPY file)", height=50)
                 
                 with gr.Row():
                     wind_speed = gr.Number(label="Wind Speed (m/s)", value=5.0)
@@ -369,29 +387,36 @@ def create_gradio_interface():
             # 출력 섹션 (2/3 크기)
             with gr.Column(scale=2):
                 output_plot = gr.Plot(label="Prediction Results")
+
+        # 예제 데이터 로드
+        examples = load_examples_from_directory("filtered_climate_data")
+        
+        # 예제 갤러리 생성
+        with gr.Row():
+            gr.Examples(
+                examples=examples,
+                inputs=[rgb_input, ndvi_input, terrain_input, elevation_input,
+                       wind_speed, wind_direction, temperature, humidity],
+                outputs=output_plot,
+                fn=predict_with_processing,
+                cache_examples=True,
+                label="Click any example to run",
+                examples_per_page=5
+            )
         
         # 예측 버튼 클릭 이벤트 연결
         predict_btn.click(
-            fn=predictor.predict_from_inputs,
+            fn=predict_with_processing,
             inputs=[rgb_input, ndvi_input, terrain_input, elevation_input,
                    wind_speed, wind_direction, temperature, humidity],
             outputs=output_plot
         )
-        
-        # 예제 섹션
-        examples = load_examples_from_directory("filtered_climate_data")
-        gr.Examples(
-            examples=examples,
-            inputs=[rgb_input, ndvi_input, terrain_input, elevation_input,
-                   wind_speed, wind_direction, temperature, humidity],
-            outputs=output_plot,
-            fn=predictor.predict_from_inputs,  # 예제 실행에 사용할 함수 지정
-            cache_examples=True,
-            api_name=False  # API 엔드포인트 생성 비활성화
-        )
     
     return interface
 
 if __name__ == "__main__":
     interface = create_gradio_interface()
-    interface.launch()
+    interface.launch(
+        share=True,
+        enable_queue=True
+    )