summarize_paper.py

import os
import dotenv
import json
from gradio_client import Client

dotenv.load_dotenv()

def summarize_paper(pdf_url, paper_id, access_key):
    mindmap = None
    summary = None
    try:
        summarizer_client = Client(
            "raannakasturi/ReXploreAPI",
            hf_token=os.getenv("HF_API_TOKEN"),
        )
        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])
            print
            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