Update app.py

app.py

@@ -1,60 +1,60 @@
-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
-
-def generate(file):
-    print(f"Generating mindmap for {file}")
-    summary = "This is a summary of the research paper"
-    mindmap_markdown = generate_markdown(llm, file)
-    mindmap_svg = generate_mindmap_svg(mindmap_markdown)
-    print("Mindmap generated successfully")
-    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', file_types=['.pdf'])
-    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, format='svg')
-    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__":
-    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 using sudo")
-    except:
-            print("Graphviz installation failed")
-            sys.exit(1)
-    print("Graphviz loaded successfully")
-    llm = load_llm_model()
-    print("Model loaded successfully")
-    app.queue(default_concurrency_limit=1).launch(show_error=True)
+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
+
+def generate(file):
+    print(f"Generating mindmap for {file}")
+    summary = "This is a summary of the research paper"
+    mindmap_markdown = generate_markdown(llm, file)
+    mindmap_svg = generate_mindmap_svg(mindmap_markdown)
+    print("Mindmap generated successfully")
+    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', file_types=['.pdf'])
+    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, format='svg')
+    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__":
+    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 using sudo")
+    except:
+            print("Graphviz installation failed")
+            sys.exit(1)
+    print("Graphviz loaded successfully")
+    llm = load_llm_model()
+    print("Model loaded successfully")
+    app.queue(default_concurrency_limit=10).launch(show_error=True)