Hide API key; hide test mode; add logo

app.py

@@ -172,13 +172,14 @@ block = gr.Blocks(css="""
 }
 """)
 with block:
-  api_key = gr.Textbox(
-      placeholder="Paste your OpenAPI API key here (sk-...)",
-      show_label=False,
-      lines=1,
-      type="password"
-  )
-  gr.Markdown("""Paste a paragraph below, then choose one of the modes to generate feedback.""")
+  # api_key = gr.Textbox(
+  #     placeholder="Paste your OpenAPI API key here (sk-...)",
+  #     show_label=False,
+  #     lines=1,
+  #     type="password"
+  # )
+  gr.HTML("""<div style="display: flex; justify-content: center; align-items: center"><a href="https://thinkcol.com/"><img src="https://thinkcol.com/thinkcol-logo.png" alt="ThinkCol" /></a></div>""")
+  gr.Markdown("""Paste a paragraph below, and then choose one of the modes to generate feedback.""")
   content = gr.Textbox(
       label="Paragraph"
   )
@@ -240,20 +241,20 @@ with block:
     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"
-    )
+  # 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()
@@ -262,7 +263,7 @@ with block:
   chain_intro = gr.State()
   chain_body1 = gr.State()
 
-  api_key.change(load_chain, [api_key], [chain, llm, chain_intro, chain_body1])
+  # api_key.change(load_chain, [api_key], [chain, llm, chain_intro, chain_body1])
 
   inputs = [content, chain]
   outputs = [output_before, output_after, changes, edited]
@@ -279,6 +280,8 @@ with block:
 
   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)
+  # submit_custom.click(run_custom, [content, llm, prompt], output_custom)
+
+  block.load(load_chain, gr.State(""), [chain, llm, chain_intro, chain_body1])
 
 block.launch(debug=True)