Update app.py

app.py

@@ -1,16 +1,14 @@
-
 import openai
 import gradio as gr
 from gradio import HuggingFaceDatasetSaver
 
+openai.api_key ='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'
 
-
-def openai_chat(prompt, api_key):
-    openai.api_key = api_key
+def openai_chat(prompt):
     completions = openai.Completion.create(
         engine="text-davinci-003",
         prompt=prompt+"The following is the prompt from teacher working in canvas infrastructure",
-        max_tokens=2000,
+        max_tokens=1024,
         temperature=0.5,
         stop=[" Human:", " AI:"]
     )
@@ -18,11 +16,13 @@ def openai_chat(prompt, api_key):
     message = completions.choices[0].text
     return message.strip()
 
-def chatbot(input, history=[], api_key=''):
-    output = openai_chat(input, api_key)
+def chatbot(input, history=[]):
+    output = openai_chat(input)
     history.append((input, output))
     return history, history
 
+saver = HuggingFaceDatasetSaver("chatbot_history.jsonl")  # Создаем объект сохранения истории
+
 api_key_input = gr.inputs.Textbox(label="Ключ OpenAI API", type="password")
 iface = gr.Interface(fn=chatbot,
                      inputs=["text", 'state', api_key_input],
@@ -35,7 +35,7 @@ iface = gr.Interface(fn=chatbot,
                      ],
                      cache_examples=False,
                      title="GPT-3 Модель: Text-davinci-003",
-                    
-                     allow_flagging="manual")
+                     allow_flagging="manual",
+                     datasets=[saver])  # Передаем saver в качестве аргумента, чтобы история сохранялась
 
 iface.launch()