Update app.py

app.py

@@ -5,7 +5,6 @@ from generate_mindmap import generate_mindmap_svg
 import gradio as gr
 import subprocess
 
-llm = load_llm_model()
 
 def generate(file):
     summary = "This is a summary of the research paper"
@@ -35,33 +34,25 @@ with gr.Blocks(theme=theme, title="Binary Biology") as app:
                 queue=True,
     )
 
-if __name__ == "__main__":
-    # run entire command: "CMAKE_ARGS="-DGGML_BLAS=ON -DGGML_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python"
-    try:
-        env = os.environ.copy()
-        env["CMAKE_ARGS"] = "-DGGML_BLAS=ON -DGGML_BLAS_VENDOR=OpenBLAS"
-        cmd = ["pip", "install", "llama-cpp-python"]
-        subprocess.run(cmd, env=env)
-    except:
-        cmd = ["pip", "install", "llama-cpp-python"]
-        subprocess.run(cmd)
+try:
+    env = os.environ.copy()
+    env["CMAKE_ARGS"] = "-DGGML_BLAS=ON -DGGML_BLAS_VENDOR=OpenBLAS"
+    cmd = ["pip", "install", "llama-cpp-python"]
+    subprocess.run(cmd, env=env)
+except:
+    cmd = ["pip", "install", "llama-cpp-python"]
+    subprocess.run(cmd)
+try:
     try:
-        try:
-            subprocess.run(['apt', 'install', '-y', 'graphviz'])
-            print("Graphviz installed successfully")
-        except:
-            subprocess.run(['sudo', 'apt', 'install', '-y', 'graphviz'])
-            print("Graphviz installed successfully")
+        subprocess.run(['apt', 'install', '-y', 'graphviz'])
+        print("Graphviz installed successfully")
     except:
+        subprocess.run(['sudo', 'apt', 'install', '-y', 'graphviz'])
+        print("Graphviz installed successfully using sudo")
+except:
         print("Graphviz installation failed")
         sys.exit(1)
-    print("Model loaded successfully")
-    llm = load_llm_model()
-    print("Model loaded successfully")
-    app.queue(default_concurrency_limit=5).launch(show_error=True)
-    # summary, markdown_mindmap, graphical_mindmap = generate("cr1c00107.pdf")
-    # print(summary)
-    # print("\n\n")
-    # print(markdown_mindmap)
-    # print("\n\n")
-    # print(graphical_mindmap)
+print("Graphviz loaded successfully")
+llm = load_llm_model()
+print("Model loaded successfully")
+app.queue(default_concurrency_limit=5).launch(show_error=True)