fix21

Strexas · May 29, 2024 · 60b4414 · 60b4414
1 parent eb4f7aa
commit 60b4414
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GUI/back-end/main.py b/GUI/back-end/main.py
@@ -61,7 +61,7 @@ def process():
 
   api_answer = answer + '\n\n'
   if '```' in api_answer and '```python' in api_answer:
-    execute = api_answer[api_answer.find('```python') + 9:api_answer.rfind('```')]
+    execute = 'from data_collection import *\n\n' + api_answer[api_answer.find('```python') + 9:api_answer.rfind('```')]
     old_stdout = sys.stdout
     redirected_output = sys.stdout = StringIO()
     exec(execute)