Update app.py

app.py

@@ -20,7 +20,7 @@ import re
 import subprocess
 from dataclasses import dataclass
 from tempfile import NamedTemporaryFile
-from typing import Dict, IO, List, cast, Tuple, Optional
+from typing import Dict, IO, List, cast, Tuple, Optional, Any
 
 import gradio as gr
 import matplotlib.pyplot as plt
@@ -98,6 +98,7 @@ def visualize(G: nx.Graph, seed: int = 0) -> plt.Figure:
     return fig
 
 
+# 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]]:
     with (NamedTemporaryFile() as graph,
@@ -156,7 +157,8 @@ def handler(file: IO[bytes], algorithm: str, seed: int) -> Tuple[pd.DataFrame, p
 
     G: nx.Graph = nx.read_edgelist(file.name, delimiter=delimiter, comments='\n', data=[('weight', float)])
 
-    mapping, reverse = {}, {}
+    mapping: Dict[Any, int] = {}
+    reverse: Dict[int, Any] = {}
 
     for i, node in enumerate(G):
         mapping[node] = i