Atualizacao app com split

app.py

@@ -5,19 +5,19 @@ import zipfile
 import os
 
 def process_onnx(uploaded_file):
-    
+    # Check if the uploaded file is a zip file
     if zipfile.is_zipfile(uploaded_file.name):
         with zipfile.ZipFile(uploaded_file.name, 'r') as zip_ref:
             zip_ref.extractall("/tmp")
-            onnx_file = zip_ref.namelist()[0]  
+            onnx_file = zip_ref.namelist()[0]  # Assuming there is only one ONNX file in the zip
             file_path = os.path.join("/tmp", onnx_file)
     else:
         file_path = uploaded_file.name
 
-    
+    # Load the ONNX model
     model = onnx.load(file_path)
 
-    
+    # Collect basic information about the model
     info = {
         "Model Name": model.graph.name,
         "Number of Nodes": len(model.graph.node),
@@ -25,7 +25,7 @@ def process_onnx(uploaded_file):
         "Nodes": []
     }
 
-    
+    # Iterate over each node (layer) to collect detailed information
     for node in model.graph.node:
         node_info = {
             "Name": node.name,
@@ -36,23 +36,23 @@ def process_onnx(uploaded_file):
         info["Nodes"].append(node_info)
         info["Architecture Summary"][node.op_type] += 1
 
-    
-    output = []
-    for key, value in info.items():
-        if key != "Nodes":
-            output.append(f"{key}: {value}")
-    output.append("\nComplete Nodes:")
-    for node in info["Nodes"]:
-        output.append(str(node))
+    # Format the summary output
+    summary_output = '\n'.join([f"{key}: {value}" for key, value in info.items() if key != "Nodes"])
 
-    return '\n'.join(output)
+    # Format the complete nodes output
+    nodes_output = "Complete Nodes:\n" + '\n'.join([str(node) for node in info["Nodes"]])
+
+    return summary_output, nodes_output
 
 iface = gr.Interface(
     fn=process_onnx,
-    inputs=gr.File(label="Upload .ONNX or .ZIP (with .onnx) File"),
-    outputs="text",
+    inputs=gr.File(label="Upload .ONNX or .ZIP File"),
+    outputs=[
+        gr.components.Textbox(label="Summary"),
+        gr.components.Textbox(label="Complete Nodes")
+    ],
     title="ONNX Model Scope",
-    description="Upload an ONNX file or a ZIP (containing an .onnx file) to extract and display its detailed information. This process can take some time depending on the size of the ONNX model."
+    description="Upload an ONNX file or a ZIP file containing an .onnx file to extract and display its detailed information. This process can take some time depending on the size of the ONNX model."
 )
 
 iface.launch(debug=True)