Update app.py

app.py

@@ -1,69 +1,67 @@
-import os
-import sys
-from generate_markdown import load_llm_model, generate_markdown
-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"
-    mindmap_markdown = generate_markdown(llm, file)
-    mindmap_svg = generate_mindmap_svg(mindmap_markdown)
-    return summary, mindmap_markdown, mindmap_svg
-
-theme = gr.themes.Soft(
-    primary_hue="purple",
-    secondary_hue="cyan",
-    neutral_hue="slate",
-    font=[gr.themes.GoogleFont('Syne'), gr.themes.GoogleFont('poppins'), gr.themes.GoogleFont('poppins'), gr.themes.GoogleFont('poppins')],
-)
-
-with gr.Blocks(theme=theme, title="Binary Biology") as app:
-    file = gr.File(file_count='single', label='Upload Research Paper PDF file')
-    summary = gr.TextArea(label='Summary', lines=5, interactive=False, show_copy_button=True)
-    markdown_mindmap = gr.Textbox(label='Mindmap', lines=5, interactive=False, show_copy_button=True)
-    graphical_mindmap = gr.Image(label='Graphical Mindmap', interactive=False, show_download_button=True)
-    submit = gr.Button(value='Submit')
-
-    submit.click(generate,
-                inputs=[file],
-                outputs=[summary, markdown_mindmap, graphical_mindmap],
-                scroll_to_output=True,
-                show_progress=True,
-                queue=True,
-    )
-
-if __name__ == "__main__":
-    # Download models
-    subprocess.run(['wget', '-o', 'model.gguf', 'https://huggingface.co/featherless-ai-quants/amazon-MegaBeam-Mistral-7B-300k-GGUF/resolve/main/amazon-MegaBeam-Mistral-7B-300k-Q6_K.gguf?download=true'])
-    # 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:
-        try:
-            subprocess.run(['apt', 'install', '-y', 'graphviz'])
-            print("Graphviz installed successfully")
-        except:
-            subprocess.run(['sudo', 'apt', 'install', '-y', 'graphviz'])
-            print("Graphviz installed successfully")
-    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)
+import os
+import sys
+from generate_markdown import load_llm_model, generate_markdown
+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"
+    mindmap_markdown = generate_markdown(llm, file)
+    mindmap_svg = generate_mindmap_svg(mindmap_markdown)
+    return summary, mindmap_markdown, mindmap_svg
+
+theme = gr.themes.Soft(
+    primary_hue="purple",
+    secondary_hue="cyan",
+    neutral_hue="slate",
+    font=[gr.themes.GoogleFont('Syne'), gr.themes.GoogleFont('poppins'), gr.themes.GoogleFont('poppins'), gr.themes.GoogleFont('poppins')],
+)
+
+with gr.Blocks(theme=theme, title="Binary Biology") as app:
+    file = gr.File(file_count='single', label='Upload Research Paper PDF file')
+    summary = gr.TextArea(label='Summary', lines=5, interactive=False, show_copy_button=True)
+    markdown_mindmap = gr.Textbox(label='Mindmap', lines=5, interactive=False, show_copy_button=True)
+    graphical_mindmap = gr.Image(label='Graphical Mindmap', interactive=False, show_download_button=True)
+    submit = gr.Button(value='Submit')
+
+    submit.click(generate,
+                inputs=[file],
+                outputs=[summary, markdown_mindmap, graphical_mindmap],
+                scroll_to_output=True,
+                show_progress=True,
+                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:
+        try:
+            subprocess.run(['apt', 'install', '-y', 'graphviz'])
+            print("Graphviz installed successfully")
+        except:
+            subprocess.run(['sudo', 'apt', 'install', '-y', 'graphviz'])
+            print("Graphviz installed successfully")
+    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)