a

app.py

@@ -170,12 +170,12 @@ def process_file(file):
 
 
 @spaces.GPU(duration=60)
-def generate(paper_text):
+def generate(paper_text, review_template):
     # messages = [
     #     {"role": "system", "content": "You are a pirate."},
     #     {"role": "user", "content": paper_text}
     # ]
-    messages = create_messages(REVIEW_FIELDS, paper_text)
+    messages = create_messages(review_template paper_text)
     input_ids = tokenizer.apply_chat_template(
         messages,
         add_generation_prompt=True,
@@ -204,18 +204,17 @@ steps = """Placeholder Description"""
 with gr.Blocks() as demo:
     title = gr.Markdown(title)
     steps = gr.Markdown(steps)
-    instr = gr.Markdown("## Upload your paper as a pdf file")
+    instr = gr.Markdown("## Upload your paper in pdf format")
     file_input = gr.File(file_types=[".pdf"], file_count="single")
-    markdown_field = gr.Textbox(label="Paper Text", max_lines=20, autoscroll=False)
-    markdown_field = gr.Textbox(label="Review Template", max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
-    # generate_button = gr.Button("Generate Review", interactive=not markdown_field)
-    generate_button = gr.Button("Generate Review")
-    file_input.upload(process_file, file_input, markdown_field)
-    # markdown_field.change(lambda text: gr.update(interactive=True) if len(text) > 1000 else gr.update(interactive=False), markdown_field, generate_button)
+    paper_text_field= gr.Textbox(label="Paper Text", max_lines=20, autoscroll=False)
+    review_template_field = gr.Textbox(label="Review Template", max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
+    generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
+    # generate_button = gr.Button("Generate Review")
+    file_input.upload(process_file, file_input, paper_text_field)
+    paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)
 
     review_field = gr.Markdown(label="Review")
-    # generate_button.click(fn=lambda: gr.update(interactive=False), inputs=None, outputs=generate_button).then(generate, markdown_field, review_field).then(fn=lambda: gr.update(interactive=True), inputs=None, outputs=generate_button)
-    generate_button.click(fn=lambda: gr.update(interactive=False), inputs=None, outputs=generate_button).then(generate, markdown_field, review_field).then(fn=lambda: gr.update(interactive=True), inputs=None, outputs=generate_button)
+    generate_button.click(fn=lambda: gr.update(interactive=False), inputs=None, outputs=generate_button).then(generate, [paper_text_field, review_template_field], review_field).then(fn=lambda: gr.update(interactive=True), inputs=None, outputs=generate_button)
     demo.title = "Paper Review Generator"