Update app.py

app.py

@@ -16,7 +16,10 @@ def geogeosolver(question:str, model:str="gpt-4o-mini"):
     input = translate(question)
     reader.main(input)
     run_ag()
-    image_path = f'{OUTPUT_FOLDER}/output.png'
+    if os.path.exists(f'{OUTPUT_FOLDER}/output.png'):
+        image_path = f'{OUTPUT_FOLDER}/output.png'
+    else:
+        image_path = f'{OUTPUT_FOLDER}/error.png'
     result_dict = read_solution()
     key_list = list(result_dict.keys())
     premises = result_dict[key_list[0]]
@@ -32,7 +35,7 @@ def clear_directory(directory_path: Union[str, Path] = OUTPUT_FOLDER) -> None:
         try:
             if path_object.is_dir():
                 shutil.rmtree(path_object)
-            else:
+            elif path_object.name != 'error.png':
                 path_object.unlink()
         except PermissionError:
             pass