Add tabs for different types of feedback

app.py

@@ -13,29 +13,79 @@ from diff_match_patch import diff_match_patch
 import collections
 from itertools import islice
 
-def load_chain(api_key, prompt_template):
+GRAMMAR_PROMPT = "Proofread for grammar and spelling without adding new paragraphs:\n{content}"
+
+INTRO_PROMPT = """These are the parts of a good introductory paragraph:
+1. Introduction
+2. The stage of human development of the main character
+3. Summary of story
+4. Thesis statement and outline of topics covered in essay
+For each part, say how confident you are (percentage) that the following paragraph has it, and put a quote of the sentence(s) that fulfil that part.
+{content}"""
+BODY_PROMPT1 = """Complete the following tasks for the following essay paragraph:
+1. Identify the topic sentence and determine whether it introduces an argument
+2. Outline a bullet list of key points
+3. Give a bullet list of parts of the paragraph that use quotes or other textual evidence from the book
+
+{content}"""
+BODY_PROMPT2 = """4. Give advice on how the topic sentence could be made stronger or clearer
+5. In a bullet list, state how each key point supports the topic (or if any doesn't support it)
+6. In a bullet list for each textual evidence, state whether the evidence is based on the book or uses outside sources
+7. Group the textual evidence from the book into different bullet lists based on which key point they support"""
+
+def load_chain(api_key):
   api_key = openai.api_key if api_key == "" or api_key.isspace() else api_key
   if api_key:
     llm = PromptLayerChatOpenAI(temperature=0, model_name="gpt-3.5-turbo", openai_api_key=api_key, pl_tags=["grammar"])
     prompt1 = PromptTemplate(
         input_variables=["content"],
-        template=prompt_template
+        template=GRAMMAR_PROMPT
     )
     chain = LLMChain(llm=llm,
                      prompt=prompt1,
                      memory=ConversationBufferMemory())
+    chain_intro = LLMChain(llm=llm,
+                           prompt=PromptTemplate(
+                               input_variables=["content"],
+                               template=INTRO_PROMPT
+                           ),
+                           memory=ConversationBufferMemory())
+    chain_body1 = LLMChain(llm=llm,
+                           prompt=PromptTemplate(
+                               input_variables=["content"],
+                               template=BODY_PROMPT1
+                           ),
+                           memory=ConversationBufferMemory())
 
-    return chain, llm
+    return chain, llm, chain_intro, chain_body1
 
 
-def run(content, chain):
+def run_diff(content, chain):
+  chain.memory.clear()
   edited = "\n".join([(chain.run(x) if should_check else x) for x, should_check in split_paragraphs(content)])
   return diff_words(content, edited) + (edited,)
 
+def run(content, chain):
+  chain.memory.clear()
+  return chain.run(content)
+
+def run_body(content, chain, llm):
+  output1 = run(content, chain)
+  output2 = run2(BODY_PROMPT2, chain, llm)
+  return output1 + "\n\n" + output2
+
+def run_custom(content, llm, prompt):
+  return LLMChain(llm=llm,
+                  prompt=PromptTemplate(
+                      input_variables=["content"],
+                      template=prompt
+                  )).run(content)
+
 def split_paragraphs(text):
-  return [(x, x != "" and not x.startswith("#") and not x.isspace()) for x in text.split("\n")]
+  # return [(x, x != "" and not x.startswith("#") and not x.isspace()) for x in text.split("\n")]
+  return [(text, True)]
 
-def sliding_window(iterable, n): # from python itertools recipes
+def sliding_window(iterable, n):
     # sliding_window('ABCDEFG', 4) --> ABCD BCDE CDEF DEFG
     it = iter(iterable)
     window = collections.deque(islice(it, n), maxlen=n)
@@ -113,7 +163,14 @@ def select_diff(evt: gr.SelectData, changes):
     # original, edited = text.split("→")
     return f"Why is it better to change [{original}] to [{edited}]?"
 
-block = gr.Blocks()
+block = gr.Blocks(css="""
+.diff-component {
+  white-space: pre-wrap;
+}
+.diff-component .textspan.hl {
+  white-space: normal;
+}
+""")
 with block:
   api_key = gr.Textbox(
       placeholder="Paste your OpenAPI API key here (sk-...)",
@@ -121,56 +178,97 @@ with block:
       lines=1,
       type="password"
   )
-  prompt = gr.Textbox(
-      label="Prompt",
-      value="Proofread for grammar and spelling:\n{content}"
-  )
+  gr.Markdown("""Paste a paragraph below, then choose one of the modes to generate feedback.""")
   content = gr.Textbox(
-      label="Content to revise"
+      label="Paragraph"
   )
-  submit = gr.Button(
-      value="Revise",
-      # variant="secondary"
-  ).style(full_width=False)
-
-  with gr.Row():
-    output_before = gr.HighlightedText(
-        label="Before",
-        color_map={
-            "-": "red",
-            # "→": "yellow",
-        }
+
+  with gr.Tab("Grammar/Spelling"):
+    gr.Markdown("Suggests grammar and spelling revisions.")
+    submit = gr.Button(
+        value="Revise",
+    ).style(full_width=False)
+
+    with gr.Row():
+      output_before = gr.HighlightedText(
+          label="Before",
+          combine_adjacent=True,
+          elem_classes="diff-component"
+      ).style(color_map={
+          "-": "red",
+          # "→": "yellow",
+      })
+      output_after = gr.HighlightedText(
+          label="After",
+          combine_adjacent=True,
+          elem_classes="diff-component"
+      ).style(color_map={
+          "+": "green",
+          # "→": "yellow",
+      })
+
+    followup_question = gr.Textbox(
+        label="Follow-up Question",
     )
-    output_after = gr.HighlightedText(
-        label="After",
-        color_map={
-            "+": "green",
-            # "→": "yellow",
-        }
+    followup_submit = gr.Button(
+        value="Ask"
+    ).style(full_width=False)
+    followup_answer = gr.Textbox(
+        label="Answer"
     )
+  with gr.Tab("Intro"):
+    gr.Markdown("Checks for the key components of an introductory paragraph.")
+    submit_intro = gr.Button(
+        value="Run"
+    ).style(full_width=False)
 
-  followup_question = gr.Textbox(
-      label="Follow-up Question",
-  )
-  followup_submit = gr.Button(
-      value="Ask"
-  ).style(full_width=False)
-  followup_answer = gr.Textbox(
-      label="Answer"
-  )
+    output_intro = gr.Textbox(
+        label="Output"
+    )
+  with gr.Tab("Body Paragraph"):
+    gr.Markdown("""1. identifies the topic sentence
+2. outlines key points
+3. checks for quotes/concrete details (e.g., textual evidence)
+4. suggests topic sentence improvements
+5. checks that the outline matches the topic
+6. checks whether the quotes/evidence are from the book
+7. checks that key points use textual evidence from the book
+8. checks for further critical analysis""")
+    submit_body = gr.Button(
+        value="Run"
+    ).style(full_width=False)
+
+    output_body = gr.Textbox(
+        label="Output"
+    )
+  with gr.Tab("Custom prompt"):
+    gr.Markdown("This mode is for testing and debugging.")
+    prompt = gr.Textbox(
+        label="Prompt",
+        value=GRAMMAR_PROMPT,
+        lines=2
+    )
+    submit_custom = gr.Button(
+        value="Run"
+    ).style(full_width=False)
+
+    output_custom = gr.Textbox(
+        label="Output"
+    )
 
   changes = gr.State()
   edited = gr.State()
   chain = gr.State()
   llm = gr.State()
+  chain_intro = gr.State()
+  chain_body1 = gr.State()
 
-  api_key.change(load_chain, [api_key, prompt], [chain, llm])
-  prompt.change(load_chain, [api_key, prompt], [chain, llm])
+  api_key.change(load_chain, [api_key], [chain, llm, chain_intro, chain_body1])
 
   inputs = [content, chain]
   outputs = [output_before, output_after, changes, edited]
-  content.submit(run, inputs=inputs, outputs=outputs)
-  submit.click(run, inputs=inputs, outputs=outputs)
+  # content.submit(run_diff, inputs=inputs, outputs=outputs)
+  submit.click(run_diff, inputs=inputs, outputs=outputs)
 
   output_before.select(select_diff, changes, followup_question)
   output_after.select(select_diff, changes, followup_question)
@@ -180,4 +278,8 @@ with block:
   followup_question.submit(run2, inputs2, outputs2)
   followup_submit.click(run2, inputs2, outputs2)
 
+  submit_intro.click(run, [content, chain_intro], output_intro)
+  submit_body.click(run_body, [content, chain_body1, llm], output_body) # body part A only
+  submit_custom.click(run_custom, [content, llm, prompt], output_custom)
+
 block.launch(debug=True)