summarize_paper.py

import json
from gradio_client import Client

def summarize_paper(pdf_url, paper_id, access_key):
    mindmap = None
    summary = None
    summarizer_client = Client("raannakasturi/ReXploreAPI")
    result = summarizer_client.predict(
        url=pdf_url,
        id=paper_id,
        access_key=access_key,
        api_name="/rexplore_summarizer"
    )
    data = json.loads(result[0])
    mindmap = data.get('mindmap')
    summary = data.get('summary')
    return summary, mindmap