Update app.py

app.py

@@ -2,21 +2,28 @@ import openai
 import os
 import gradio as gr
 openai.api_key = os.environ['openai']
-def write_essay(prompt, temperature, frequency_penalty, presence_penalty, max_token, model="gpt-3.5-turbo"):
+def write_essay(prompt, context, temperature, frequency_penalty, presence_penalty):
+    model="gpt-3.5-turbo"
     temp_list = prompt.split()
+    temp_list2 = context.split()
     temperature = temperature/10
     frequency_penalty = frequency_penalty/10 - 2
     presence_penalty = presence_penalty/10 -2
-    if len(temp_list) >= 500:
-        new_prompt = " ".join(temp_list[0:500])
+    if len(temp_list) >= 100:
+        new_prompt = " ".join(temp_list[0:100])
     else:
         new_prompt = prompt
+
+    if len(temp_list2) >= 300:
+        new_prompt2 = " ".join(temp_list2[0:300])
+    else:
+        new_prompt2 = context
     
     response = openai.ChatCompletion.create(
         model=model,
-        messages=[{"role": "user", "content": new_prompt}],
-        #prompt=new_prompt,
-        max_tokens=int(max_token),
+        messages=[{"role": "system", "content": new_prompt2},
+                  {"role": "user", "content": new_prompt},],
+        max_tokens=400,
         temperature=temperature,
         frequency_penalty=frequency_penalty,
         presence_penalty=presence_penalty
@@ -25,8 +32,9 @@ def write_essay(prompt, temperature, frequency_penalty, presence_penalty, max_to
     message = response.choices[0]["message"]["content"]
     #print(message)
     return message
-demo = gr.Interface(fn=write_essay, inputs=["text",gr.Slider(0, 20, value=4, step=1, label="Creative vs Focus", info="Increase for creative and decrease for more focused"),
+demo = gr.Interface(fn=write_essay, inputs=["text","text",
+                                            gr.Slider(0, 20, value=4, step=1, label="Creative vs Focus", info="Increase for creative and decrease for more focused"),
                                             gr.Slider(0, 40, value=20, step=1, label="Frequency of words", info="Increase to decrease repeated words"),
-                                            gr.Slider(0, 40, value=20, step=1, label="Presence of words", info="Increase to decrease already appeared words"),"text"
+                                            gr.Slider(0, 40, value=20, step=1, label="Presence of words", info="Increase to decrease already appeared words")
                                            ], outputs="text")
-demo.launch()
+demo.launch()