Upload 2 files

app.py

@@ -0,0 +1,116 @@
+
+from gradio_client import Client
+import json
+import requests
+import gradio as gr
+import ast
+
+def upload_pdf(pdf_path):
+    api_url = "https://tmpfiles.org/api/v1/upload"
+    with open(pdf_path, 'rb') as file:
+        files = {'file': file}
+        response = requests.post(api_url, files=files)
+    url = response.json()['data']['url']
+    download_url = f"https://tmpfiles.org/dl/{url.split('.org/')[-1]}"
+    return download_url
+
+def fetch_doi_data(pdf_path):
+    pdf_url = upload_pdf(pdf_path)
+    client = Client("raannakasturi/ScientryPDFDataAPI")
+    result = client.predict(pdf_url=pdf_url, api_name="/getDOIData")
+    result = json.loads(result)
+    return pdf_url, result
+
+def generate_summary_mindmap(pdf_url, doi):
+    client = Client("raannakasturi/ScientryAPI")
+    result = client.predict(
+        url=pdf_url,
+        id=doi,
+        access_key="scientrypass",
+        api_name="/rexplore_summarizer"
+    )
+    return result
+
+def generate_mindmap(markdown_mindmap):
+    client = Client("raannakasturi/MindMap")
+    result = client.predict(
+            mindmap_markdown=markdown_mindmap,
+            api_name="/generate"
+    )
+    return result
+
+def create_files(title, content, file_type):
+    file_name = f"{title}.{file_type}"
+    if isinstance(content, str) and content.startswith("b'"):
+        content = ast.literal_eval(content)
+    with open(file_name, "wb") as file:
+        file.write(content)
+    return file_name
+
+
+def main(pdf_path):
+    pdf_url, doi_data = fetch_doi_data(pdf_path)
+    summary_mindmap = generate_summary_mindmap(pdf_url, doi_data["doi"])
+    markdown_summary = summary_mindmap[1]
+    markdown_mindmap = summary_mindmap[2]
+    result = generate_mindmap(markdown_mindmap)
+    if not result:
+        mindmap_svg_file = ""
+        mindmap_pdf_file = ""
+    else:
+        mindmap_svg_file = create_files(doi_data["title"], result[0], "svg")
+        mindmap_pdf_file = create_files(doi_data["title"], result[1], "pdf")
+    return doi_data["citation_text"], doi_data["title"], markdown_summary, mindmap_svg_file, mindmap_pdf_file
+
+def generate_pdf_summary_mindmap(pdf):
+    pdf_path = pdf.name
+    citation, title, markdown_summary, mindmap_svg_file, mindmap_pdf_file = main(pdf_path)
+    return citation, title, markdown_summary, mindmap_svg_file, mindmap_pdf_file
+
+def download_file(download_as_pdf):
+    return download_as_pdf
+
+with gr.Blocks(title="Scientry Local App") as app:
+    gr.HTML("""
+    <div
+        style="display: flex; padding-top: 2.5rem; flex-direction: column; justify-content: center; align-items: center; width: 100%;">
+        <div
+            style="display: flex; margin-left: 0.5rem; flex-direction: row; justify-content: center; align-items: center;">
+            <img src="https://raw.githubusercontent.com/RaannaKasturi/ScientryLocal/refs/heads/main/icon.png"
+                style="margin-right: 0.5rem" alt="ScientryIcon" width="32" height="32">
+            <h1 style="font-size: 1.875rem; line-height: 2.25rem; font-weight: 700;">Scientry (Local)</h1>
+        </div>
+        <p style="font-style: italic;">Science Simplifed, Knowledge Amplified</p>
+        <br>
+        <p>
+            Designed and Developed by
+            <a style="text-decoration: underline; text-underline-offset: 2px;" href="http://nayankasturi.eu.org"
+                target="_blank" rel="noopener noreferrer">
+                Nayan Kasturi
+            </a>
+            and
+            <a style="text-decoration: underline; text-underline-offset: 2px;" href="http://binarybiology.top"
+                target="_blank" rel="noopener noreferrer">
+                Binary Biology
+            </a>
+        </p>
+    </div>
+    """)
+    with gr.Column():
+        with gr.Row():
+            with gr.Column():
+                pdf = gr.File(file_count='single', file_types=[".pdf"], label="Upload PDF")
+                generate = gr.Button(value="Generate")
+            with gr.Column():
+                title = gr.Textbox(label="Title", placeholder="Title will be displayed here", show_copy_button=True)
+                citation = gr.Textbox(label="Citation", placeholder="Citation will be displayed here", show_copy_button=True)
+        with gr.Row():
+            summary = gr.Textbox(label="Summary", placeholder="Summary will be displayed here", interactive=True, show_copy_button=True)
+            with gr.Column():
+                download_as_svg = gr.DownloadButton(label="Download as SVG")
+                download_as_pdf = gr.DownloadButton(label="Download as PDF")
+    generate.click(generate_pdf_summary_mindmap, inputs=[pdf], outputs=[citation, title, summary, download_as_svg, download_as_pdf])
+    download_as_svg.click(download_file, inputs=[download_as_svg], outputs=[download_as_svg])
+    download_as_pdf.click(download_file, inputs=[download_as_pdf], outputs=[download_as_pdf])
+
+app.launch(inbrowser=True, show_api=False)

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+gradio-client
+requests