Update app.py

add history log file

app.py

@@ -1,32 +1,62 @@
+# example of chat with openAI
+
 import gradio as gr
 import openai
+import datetime
 import os
 
-# Set up the OpenAI API credentials
-# openai.api_key = os.environ["OPENAI_API_KEY"]
-openai.api_key = os.getenv("OPENAI_API_KEY")
-model = "gpt-3.5-turbo"
-
-# Define the chatbot function
-def chatbot(text):
+# openAI Python program guide
+# https://github.com/openai/openai-python
 
-    # Call OpenAI API to generate text completion
-    completion = openai.ChatCompletion.create(model=model, messages=[{"role": "user", "content": text}])
-
-    # Extract the generated response and return it
-    return completion.choices[0].message.content
+# 设置 OpenAI API 密钥
+openai.api_key = os.getenv("OPENAI_API_KEY")
+MODEL = "gpt-3.5-turbo"
 
+# 文件名
+FILE_NAME = "chat_history.log"
 
-def ask_gpt(prompt):
+# 定义对话函数
+def chat(question):
     try:
-        completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": prompt}])
-        return completion.choices[0].message.content
+        # 发送 API 请求
+        completion = openai.ChatCompletion.create(
+            model=MODEL, 
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": question},
+            ],
+            temperature=0.8,
+            )
+
+        response = completion.choices[0].message.content
+    
     except openai.Error as e:
-        return f"OpenAI API error: {e}"
-        
+        response = f"OpenAI API error: {e}"
     
-# Create the Gradio interface
-iface = gr.Interface(fn=ask_gpt, inputs="text", outputs="text", title="GPT-3.5 Turbo Chatbot")
+    # 获取当前日期和时间
+    now = datetime.datetime.now()
+
+    # 将日期和时间转换为字符串格式
+    date_string = now.strftime('%Y-%m-%d %H:%M:%S')
+
+    # 将提问和回答保存到聊天历史记录中
+    # 打开文件进行追加
+    with open(FILE_NAME, 'a') as f:
+        f.write(f'\n{date_string}\n')
+        f.write('You: ' + question + '\n')
+        f.write('chatGPT: ' + response + '\n')
 
-# Launch the interface
-iface.launch()
+    return response
+
+
+if __name__ == '__main__':
+    # 创建 Gradio 应用程序界面
+    iface = gr.Interface(
+        fn=chat,
+        inputs="text",
+        outputs='text',
+        title="Chat with OpenAI 3.5",
+        #description="Talk to an AI powered by OpenAI's GPT language model.",
+    )
+    
+    iface.launch(share=True)