Upload app.py

app.py

@@ -1,10 +1,17 @@
 import gradio as gr
 from reader import Reader
 from utils import *
+import shutil
+from pathlib import Path
+from typing import Union
 
 OUTPUT_FOLDER = 'ag4mout'
 
 def geogeosolver(question:str, model:str="gpt-4o-mini"):
+    if model == 'gpt-4o-mini':
+        get_github_token()
+    elif model == 'gemma2-9b-it':
+        get_grog_api()
     reader = Reader(model=model)
     input = translate(question)
     reader.main(input)
@@ -17,6 +24,20 @@ def geogeosolver(question:str, model:str="gpt-4o-mini"):
     steps = result_dict[key_list[2]]
     return image_path, premises, constructions, steps
 
+def clear_directory(directory_path: Union[str, Path] = OUTPUT_FOLDER) -> None:
+    """Irreversibly removes all files and folders inside the specified
+    directory. Returns a list with paths Python lacks permission to delete."""
+    # erroneous_paths = []
+    for path_object in Path(directory_path).iterdir():
+        try:
+            if path_object.is_dir():
+                shutil.rmtree(path_object)
+            else:
+                path_object.unlink()
+        except PermissionError:
+            pass
+            # erroneous_paths.append(path_object)
+
 if __name__ == '__main__':
     with gr.Blocks() as demo:
         with gr.Row():
@@ -28,7 +49,7 @@ if __name__ == '__main__':
                 steps = gr.Textbox(lines=3, label="Các Bước Giải")
             with gr.Column():
                 image = gr.Image(type="filepath", label="Hình Vẽ")
-        solve_button.click(geogeosolver, inputs=question, outputs=[image, premises, constructions, steps])
+        solve_button.click(geogeosolver, inputs=question, outputs=[image, premises, constructions, steps]).then(clear_directory)
         example = gr.Examples(
             examples = [
             'Cho tam giác ABC nhọn, vẽ các đường cao AD, BE, CF. Gọi H là trực tâm của tam giác. Gọi M, N, P, Q lần lượt là các hình chiếu vuông góc của D lên AB, BE, CF, AC. Chứng minh: tứ giác BMND nội tiếp.',
@@ -36,4 +57,4 @@ if __name__ == '__main__':
             ],
             inputs = [question],
             )
-    demo.launch(share=True)
+    demo.launch()