do some field stuff

app.py

@@ -28,6 +28,23 @@ examples = [
 ]
 
 
+# Example prompt
+#   "input": "```\n_BOOL8 __fastcall sub_409B9A(_QWORD *a1, _QWORD *a2)\n{\nreturn *a1 < *a2 || *a1 == *a2 && a1[1] < a2[1];\n}\n```\nWhat are the variable name and type for the following memory accesses:a1, a1[1], a2, a2[1]?\n",
+#  "output": "a1: a, os_reltime* -> sec, os_time_t\na1[1]: a, os_reltime* -> usec, os_time_t\na2: b, os_reltime* -> sec, os_time_t\na2[1]: b, os_reltime* -> usec, os_time_t",
+def field_prompt(code):
+    field_helper_result = gradio_client.predict(
+        decompiled_code=code,
+        api_name="/predict",
+    )
+    print(f"field helper result: {field_helper_result}")
+
+    fields = sorted([e['expr'] for e in field_helper_result[0] if e['expr'] != ''])
+    print(f"fields: {fields}")
+
+    prompt = f"```\n{code}\n```\nWhat are the variable name and type for the following memory accesses:{', '.join(fields)}?\n"
+
+    return prompt, field_helper_result
+
 @spaces.GPU
 def infer(code):
 
@@ -72,11 +89,7 @@ def infer(code):
         clean_up_tokenization_spaces=True,
     )
 
-    field_helper_result = gradio_client.predict(
-        decompiled_code=code,
-        api_name="/predict",
-    )
-    print(field_helper_result)
+    field_prompt_result, field_helper_result = field_prompt(code)
 
     # field_output = fielddecoder_model.generate(
     #     input_ids=input_ids,