more processing

app.py

@@ -20,6 +20,12 @@ This space simply performs inference on the two pretrained models available as
 part of the ReSym artifacts. It takes a variable name and some decompiled code
 as input, and outputs the variable type and other information.
 
+## Disclaimer
+
+I'm not a ReSym developer and I may have messed something up.  In particular,
+you must prompt the variable names in the decompiled code as part of the prompt,
+and I reused some of their own code to do this.
+
 ## Todo
 
 * Add support for FieldDecoder model
@@ -57,9 +63,12 @@ example = r"""{
 def infer(var_name, code):
 
     splitcode = code.splitlines()
-    comments = prep_decompiled.extract_comments(splitcode)
-    sig = prep_decompiled.parse_signature(splitcode)
-    print(f"comments={comments} sig={sig}")
+    bodyvars = [v.name for v in prep_decompiled.extract_comments(splitcode) if hasattr(v, "name")]
+    argvars = [v.name for v in prep_decompiled.parse_signature(splitcode) if hasattr(v, "name")]
+    vars = argvars + bodyvars
+    #comments = prep_decompiled.extract_comments(splitcode)
+    #sig = prep_decompiled.parse_signature(splitcode)
+    print(f"vars {vars}")
 
     #line = json.loads(input)
     #first_token = line["output"].split(":")[0]