ejschwartz commited on
Commit
d469baf
·
1 Parent(s): c51e84d

Try to call

Browse files
Files changed (1) hide show
  1. main.py +24 -2
main.py CHANGED
@@ -1,7 +1,29 @@
 
1
  import gradio as gr
 
 
2
 
3
- def predict(inp):
4
- return "Hi"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
 
7
  def run():
 
1
+ import tempfile
2
  import gradio as gr
3
+ import os
4
+ import subprocess
5
 
6
+ def predict(decompiled_code):
7
+
8
+ with tempfile.TemporaryDirectory() as temp_dir:
9
+
10
+ code_file_name = os.path.join(temp_dir, "code.decompiled")
11
+
12
+ with open(code_file_name, "w") as f:
13
+ f.write(decompiled_code)
14
+
15
+ file_save_dir = os.path.join(temp_dir, "file_save_dir")
16
+ parsed_save_dir = os.path.join(temp_dir, "parsed_save_dir")
17
+
18
+ os.makedirs(file_save_dir, exist_ok=True)
19
+ os.makedirs(parsed_save_dir, exist_ok=True)
20
+
21
+ subprocess.check_output(["python", "/home/ReSym/process_data/prep_decompiled.py", code_file_name, file_save_dir, parsed_save_dir])
22
+
23
+ os.system(f"ls {file_save_dir}")
24
+ os.system(f"ls {parsed_save_dir}")
25
+
26
+ return "Almost"
27
 
28
 
29
  def run():