allow modifying prompts

app.py

@@ -10,9 +10,9 @@ The essence of the text should remain unaltered, including any LaTeX commands.
 I request that you provide only the improved version of the text without any further explanations.
 """
 
-PREFIX_PROMPT = "Please refine the following text in academic English: \n"
+PREFIX_PROMPT = "Please refine the following text in academic English:"
 
-def submit(x, api, simple=False):
+def submit(x, api, init_prompt, prefix_prompt, simple=False):
     # reset api key everytime, so it won't save api unsafely...?
     openai.api_key = api 
     # restart a new conversation.
@@ -20,15 +20,15 @@ def submit(x, api, simple=False):
         results = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=[
-                {"role": "user", "content": PREFIX_PROMPT + x},
+                {"role": "user", "content": prefix_prompt + '\n' + x},
             ]
         )
     else:    
         results = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=[
-                {"role": "system", "content": INIT_PROMPT},
-                {"role": "user", "content": PREFIX_PROMPT + x},
+                {"role": "system", "content": init_prompt},
+                {"role": "user", "content": prefix_prompt + '\n' + x},
             ]
         )
     total_tokens = results['usage']['total_tokens']
@@ -42,20 +42,23 @@ with gr.Blocks() as app:
     gr.Markdown("### ChatGPT, please help to improve my paper writing!")
 
     # allow setting API key in gui
-    with gr.Row():
-        api_input = gr.Textbox(label="OPENAI_API_KEY", value=openai.api_key, lines=1)
+    api_input = gr.Textbox(label="OPENAI_API_KEY", value=openai.api_key, lines=1)
+
+    # allow changing prompts
+    init_prompt_input = gr.Textbox(label="Init Prompt", value=INIT_PROMPT)
+    prefix_prompt_input = gr.Textbox(label="Prefix Prompt", value=PREFIX_PROMPT)
 
     simple_checkbox = gr.Checkbox(value=False, label='simple mode (only send the prefix prompt, 0.00142$ cheaper per query)')
 
     with gr.Row():
 
-        text_input = gr.Textbox(label="Input", lines=10)
+        text_input = gr.Textbox(label="Input", lines=12)
 
         with gr.Column():
             text_output = gr.Textbox(label="Output", lines=10)
             cost = gr.Number(label='cost of this query ($)')
 
     text_button = gr.Button("Submit")
-    text_button.click(submit, inputs=[text_input, api_input, simple_checkbox], outputs=[text_output, cost])
+    text_button.click(submit, inputs=[text_input, api_input, init_prompt_input, prefix_prompt_input, simple_checkbox], outputs=[text_output, cost])
     
 app.launch()