Update app.py

app.py

@@ -23,7 +23,7 @@ import re
 import subprocess
 from dataclasses import dataclass
 from tempfile import NamedTemporaryFile
-from typing import Dict, IO, List, cast, Tuple, Optional, Any
+from typing import cast, BinaryIO, Optional
 
 import gradio as gr
 import matplotlib.pyplot as plt
@@ -47,7 +47,7 @@ class Algorithm:
     global_params: Optional[str] = None
     bin: Optional[str] = None
 
-    def args_clustering(self) -> List[str]:
+    def args_clustering(self) -> list[str]:
         args = [self.name]
 
         if self.mode:
@@ -66,7 +66,7 @@ class Algorithm:
 
         return args
 
-    def args_graph(self) -> List[str]:
+    def args_graph(self) -> list[str]:
         args = []
 
         if self.local_name:
@@ -78,7 +78,7 @@ class Algorithm:
         return args
 
 
-ALGORITHMS: Dict[str, Algorithm] = {
+ALGORITHMS: dict[str, Algorithm] = {
     'CW_top': Algorithm('cw', 'top'),
     'CW_lin': Algorithm('cw', 'lin'),
     'CW_log': Algorithm('cw', 'log'),
@@ -122,8 +122,8 @@ def visualize(G: nx.Graph, seed: int = 0) -> Figure:
 
 
 # noinspection PyPep8Naming
-def watset(G: nx.Graph, algorithm: str, seed: int = 0,
-           jar: str = 'watset.jar', timeout: int = 10) -> Tuple[pd.DataFrame, Optional[nx.Graph]]:
+def watset(G: nx.Graph[str], algorithm: str, seed: int = 0,
+           jar: str = 'watset.jar', timeout: int = 10) -> tuple[pd.DataFrame, Optional[nx.Graph[str]]]:
     with (NamedTemporaryFile() as graph,
           NamedTemporaryFile(mode='rb') as clusters,
           NamedTemporaryFile(mode='rb') as senses):
@@ -164,7 +164,7 @@ def watset(G: nx.Graph, algorithm: str, seed: int = 0,
         return df_clusters, None
 
 
-def handler(file: IO[bytes], algorithm: str, seed: int) -> Tuple[pd.DataFrame, Figure]:
+def handler(file: BinaryIO, algorithm: str, seed: int) -> tuple[pd.DataFrame, Figure]:
     if file is None:
         raise gr.Error('File must be uploaded')
 
@@ -178,10 +178,10 @@ def handler(file: IO[bytes], algorithm: str, seed: int) -> Tuple[pd.DataFrame, F
         except csv.Error:
             delimiter = ','
 
-    G: nx.Graph = nx.read_edgelist(file.name, delimiter=delimiter, comments='\n', data=[('weight', float)])
+    G: nx.Graph[str] = nx.read_edgelist(file.name, delimiter=delimiter, comments='\n', data=[('weight', float)])
 
-    mapping: Dict[Any, int] = {}
-    reverse: Dict[int, Any] = {}
+    mapping: dict[str, int] = {}
+    reverse: dict[int, str] = {}
 
     for i, node in enumerate(G):
         mapping[node] = i
@@ -217,7 +217,7 @@ def main() -> None:
                 label='Graph'
             ),
             gr.Dropdown(
-                choices=cast(List[str], ALGORITHMS),
+                choices=cast(list[str], ALGORITHMS),
                 value='Watset[MCL, CW_lin]',
                 label='Algorithm'
             ),