summarize_paper.py

import json
from gradio_client import Client

def summarize_paper(pdf_url, paper_id, access_key):
    mindmap = None
    summary = None
    try:
        summarizer_client = Client("raannakasturi/ReXploreAPI")
        result = summarizer_client.predict(
            url=pdf_url,
            id=paper_id,
            access_key=access_key,
            api_name="/rexplore_summarizer"
        )
        if result:
            data = json.loads(result[0])
            if data["mindmap_status"] == "success":
                mindmap = data["mindmap"]
            if data["summary_status"] == "success":
                summary = data["summary"]
    except Exception as e:
        print(f"Error summarizing paper: {e}")
    return summary, mindmap