|
|
import os |
|
|
import gradio as gr |
|
|
from suggest import Suggest |
|
|
from edit import Editor |
|
|
from config import configure_logging |
|
|
|
|
|
|
|
|
configure_logging() |
|
|
|
|
|
|
|
|
with gr.Blocks() as demo: |
|
|
|
|
|
title = gr.Button("PaperGPT", interactive=False) |
|
|
key = gr.Textbox(label="openai_key", value=os.environ.get('OPENAI_API_KEY')) |
|
|
|
|
|
with gr.Tab("Edit"): |
|
|
|
|
|
handler = Editor() |
|
|
txt_in = gr.Textbox(label="Input", lines=11, max_lines=11, value=handler.sample_content) |
|
|
btn = gr.Button("Edit") |
|
|
txt_out = gr.Textbox(label="Output", lines=11, max_lines=11, value="GPT will serve as your editor and modify the paragraph for you.") |
|
|
btn.click(handler.generate, inputs=[txt_in, key], outputs=[txt_out]) |
|
|
|
|
|
with gr.Tab("Suggest"): |
|
|
|
|
|
idea_list = [] |
|
|
max_ideas = 20 |
|
|
handler = Suggest(max_ideas) |
|
|
|
|
|
def select(name: str): |
|
|
global idea_list |
|
|
for i in idea_list: |
|
|
if i['title'] == name: |
|
|
return [ |
|
|
gr.Textbox.update(value=i["thought"], label="thought", visible=True), |
|
|
gr.Textbox.update(value=i["action"], label="action", visible=True), |
|
|
gr.Textbox.update(value=i["original"], label="original", visible=True, max_lines=5, lines=5), |
|
|
gr.Textbox.update(value=i["improved"], label="improved", visible=True, max_lines=5, lines=5) |
|
|
] |
|
|
|
|
|
with gr.Row().style(equal_height=True): |
|
|
with gr.Column(scale=0.95): |
|
|
txt_in = gr.Textbox(label="Input", lines=11, max_lines=11, value=handler.sample_content[2048+2048+256-45:]) |
|
|
with gr.Column(scale=0.05): |
|
|
upload = gr.File(file_count="single", file_types=["tex", ".pdf"]) |
|
|
btn = gr.Button("Analyze") |
|
|
upload.change(handler.read_file, inputs=upload, outputs=txt_in) |
|
|
|
|
|
textboxes = [] |
|
|
sug = gr.Textbox("GPT will give suggestions and help you improve the paper quality.", interactive=False, show_label=False).style(text_align="center") |
|
|
with gr.Row(): |
|
|
with gr.Column(scale=0.4): |
|
|
for i in range(max_ideas): |
|
|
t = gr.Button("", visible=False) |
|
|
textboxes.append(t) |
|
|
with gr.Column(scale=0.6): |
|
|
thought = gr.Textbox(label="thought", visible=False, interactive=False) |
|
|
action = gr.Textbox(label="action", visible=False, interactive=False) |
|
|
original = gr.Textbox(label="original", visible=False, max_lines=5, lines=5, interactive=False) |
|
|
improved = gr.Textbox(label="improved", visible=False, max_lines=5, lines=5, interactive=False) |
|
|
|
|
|
btn.click(handler.generate, inputs=[txt_in, key], outputs=[sug, btn, thought, action, original, improved] + textboxes) |
|
|
for i in textboxes: |
|
|
i.click(select, inputs=[i], outputs=[thought, action, original, improved]) |
|
|
|
|
|
|
|
|
|
|
|
demo.launch(enable_queue=True) |
|
|
|
