Update app.py

app.py

@@ -2,11 +2,14 @@ import gradio as gr
 import networkx as nx
 import pydot
 import pandas as pd
-from io import BytesIO
 
-def calculate_parameters(file):
-    # Parse dot file using pydot
-    graphs = pydot.graph_from_dot_file(file.name)
+def calculate_parameters(dot_file):
+    # Read the dot file with specified encoding
+    with open(dot_file.name, 'r', encoding='utf-8') as f:
+        dot_data = f.read()
+
+    # Parse dot data using pydot
+    graphs = pydot.graph_from_dot_data(dot_data)
     G = nx.nx_pydot.from_pydot(graphs[0])
 
     # Initialize the list of lengths and the node-to-index map
@@ -38,15 +41,17 @@ def calculate_parameters(file):
     # Calculate average width (Wavg)
     Wavg = Wabs / len(set(all_lengths))
 
-    # Create a DataFrame for node depths
+    # Create a DataFrame for better visualization
     df = pd.DataFrame.from_dict(node_depth, orient='index', columns=['Depth'])
-    node_depth_str = df.to_string()
-
-    result = f"Node Depths:\n{node_depth_str}\n\nFinal Calculations:\n"
-    result += f"Dabs = {Dabs}, Dmax = {Dmax}, Davg = {Davg:.3f}\n"
-    result += f"Wabs = {Wabs}, Wmax = {Wmax}, Wavg = {Wavg:.3f}"
+    df_str = df.to_string()
 
-    return result
+    result_str = (f"Node Depths:\n{df_str}\n\nFinal Calculations:\n"
+                  f"Dabs = {Dabs}, Dmax = {Dmax}, Davg = {Davg:.3f}\n"
+                  f"Wabs = {Wabs}, Wmax = {Wmax}, Wavg = {Wavg:.3f}")
+    
+    return result_str
 
-iface = gr.Interface(fn=calculate_parameters, inputs="file", outputs="text")
+iface = gr.Interface(fn=calculate_parameters,
+                     inputs=gr.inputs.File(label="Upload .dot file"),
+                     outputs="text")
 iface.launch()