model saving

app.py

@@ -158,19 +158,23 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
+
 import gradio as gr
 import os
 import subprocess
-from pathlib import Path
 
 # Function to visualize the model
 def visualize_model(model_file):
+    # Create the output directory if it doesn't exist
+    output_dir = os.path.join("./models", os.path.dirname(model_file.name))
+    os.makedirs(output_dir, exist_ok=True)  # FIXED: Create nested directories if needed
+
     # Save the uploaded file
-    file_path = f"./models/{model_file.name}"
+    file_path = os.path.join("./models", model_file.name)
     with open(file_path, "wb") as f:
-        f.write(model_file.data)  # FIXED: Use 'data' instead of 'read()'
+        f.write(model_file.data)  # Save the uploaded file data
 
-    # Start Netron server to serve the file
+    # Start Netron server to visualize the model
     port = 8081
     subprocess.Popen(
         ["netron", "--host", "0.0.0.0", "--port", str(port), file_path],
@@ -183,9 +187,6 @@ def visualize_model(model_file):
     return f"Model uploaded successfully! View it at: {netron_url}"
 
 
-# Create output directory
-os.makedirs("./models", exist_ok=True)
-
 # Gradio Interface
 demo = gr.Interface(
     fn=visualize_model,
@@ -195,5 +196,5 @@ demo = gr.Interface(
     description="Upload a model file to visualize its architecture using Netron.",
 )
 
-# Launch the app with public link support
+# Launch the app
 demo.launch(server_name="0.0.0.0", server_port=7860, share=True)