import gradio as gr
import pandas as pd
from util import summarizer, parser

def patent_summarizer(patent_information, summaries_generated, min_char_abs, min_char_bg, min_char_claims):
    
    # TODO: add checker if valid patent info, return None if invalid
    # TODO: add scraper to get document

    # TODO: add parser to get the following info from the base document:
    abstract, background, claims = None, None, None

    summaries = list()
    if "Abstract" in summaries_generated:
        abstract_summary = summarizer.generate_abs_summary(abstract, min_char_abs)
        summaries.append(abstract_summary)
    else:
        summaries.append(None)
  
    if "Background" in summaries_generated:
        background_summary = summarizer.generate_bg_summary(background, min_char_bg)
        summaries.append(background_summary)
    else:
        summaries.append(None)
  
    if "Claims" in summaries_generated:
        claims_summary = summarizer.generate_claims_summary(claims, min_char_claims)
        summaries.append(claims_summary)
    else:
        summaries.append(None)

    return summaries

summary_options = ["Abstract", "Background", "Claims"]
iface = gr.Interface(
    patent_summarizer,
    theme="dark-peach",
    examples=[
        [
            "US9820315B2",
            summary_options,
            50,
            250,
            100
        ],
        [
            "https://patents.google.com/patent/US9820315B2",
            summary_options,
            50,
            250,
            100
        ],
        [
            "https://patents.google.com/patent/US10263802B2/en?q=smart+user+interface&oq=smart+user+interface",
            summary_options[:-1],
            75,
            200,
            125
            
        ]
    ],
    inputs=[
        gr.inputs.Textbox(label="Patent Information", 
                          placeholder="US9820315B2 or https://patents.google.com/patent/US9820315B2 ...", 
                          lines=1),
        # gr.inputs.Radio(["Link", "Patent Code"]), # Can be figured out automatically via parsing
        # gr.inputs.Dropdown(["model type 1", "model type 2", "model type 3"]), # Change model type
        gr.inputs.CheckboxGroup(summary_options, default=summary_options),
        gr.inputs.Slider(minimum=50,
                         maximum=500,
                         step=1,
                         default=75,
                         label="Minimum Characters (Abstract Summary)"),
        gr.inputs.Slider(minimum=50,
                         maximum=500,
                         step=1,
                         default=200,
                         label="Minimum Characters (Background Summary)"),
        gr.inputs.Slider(minimum=50,
                         maximum=500,
                         step=1,
                         default=100,
                         label="Minimum Characters (Per Claims Summary)")
    ],
    outputs=[
        gr.outputs.Textbox(label="Abstract Summary"),
        gr.outputs.Textbox(label="Background Summary"),
        gr.outputs.Textbox(label="Claims Summary")
    ],
    title="Patent Summarizer v.1.0.0",
    description="""
    """
)

iface.launch()