sudo-soldier commited on
Commit
d3acb13
·
verified ·
1 Parent(s): fbd48b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -5,12 +5,11 @@ def load_mesh_with_info(mesh_file):
5
  if not mesh_file:
6
  return None, "No file selected"
7
 
8
- file_size = os.path.getsize(mesh_file) / 1024 # Convert to KB
9
  return mesh_file, f"File: {os.path.basename(mesh_file)}, Size: {file_size:.2f} KB"
10
 
11
  files_dir = "files"
12
 
13
-
14
  model_files = [
15
  "model1.glb",
16
  "model2.glb",
@@ -18,7 +17,6 @@ model_files = [
18
  "model4.glb"
19
  ]
20
 
21
-
22
  for model_file in model_files:
23
  file_path = os.path.join(files_dir, model_file)
24
  if not os.path.exists(file_path):
@@ -39,8 +37,13 @@ with gr.Blocks() as demo:
39
 
40
  model_output = gr.Model3D(clear_color=[0.1, 0.1, 0.1, 1.0], label="Preview Model")
41
 
42
- model_input.change(load_mesh_with_info, inputs=model_input, outputs=[model_output, file_info])
 
 
 
43
 
 
 
44
 
45
  examples = gr.Examples(
46
  examples=[
@@ -63,3 +66,4 @@ if __name__ == "__main__":
63
 
64
 
65
 
 
 
5
  if not mesh_file:
6
  return None, "No file selected"
7
 
8
+ file_size = os.path.getsize(mesh_file) / 1024
9
  return mesh_file, f"File: {os.path.basename(mesh_file)}, Size: {file_size:.2f} KB"
10
 
11
  files_dir = "files"
12
 
 
13
  model_files = [
14
  "model1.glb",
15
  "model2.glb",
 
17
  "model4.glb"
18
  ]
19
 
 
20
  for model_file in model_files:
21
  file_path = os.path.join(files_dir, model_file)
22
  if not os.path.exists(file_path):
 
37
 
38
  model_output = gr.Model3D(clear_color=[0.1, 0.1, 0.1, 1.0], label="Preview Model")
39
 
40
+ def handle_model_input(selected_model):
41
+ if selected_model:
42
+ print(f"Loading model: {selected_model}")
43
+ return selected_model
44
 
45
+ model_input.change(handle_model_input, inputs=model_input, outputs=model_output)
46
+ model_input.change(load_mesh_with_info, inputs=model_input, outputs=[model_output, file_info])
47
 
48
  examples = gr.Examples(
49
  examples=[
 
66
 
67
 
68
 
69
+