Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
-messages = [{"role": "system", "content": """Please create engaging and informative video scripts for real 
+initial_messages = [{"role": "system", "content": """Please create engaging and informative video scripts for real 
 estate agents to use on social media. The target audience is potential homebuyers and sellers.
 The tone should be professional and friendly, with a focus on building trust and showcasing the agent's expertise. 
 Please incorporate a strong call-to-action at the end of each script, 
@@ -13,7 +13,7 @@ Write only the words the agent should read into the camera. Once you create the
 engaging hooks that could replace the first sentence. Analyze those 5 hooks, pick the best one, and replace the 
 existing hook in the script you wrote with the new one. Deliver the script with the new hook included."""}]
 
-def CustomChatGPT(user_input):
+def CustomChatGPT(user_input, messages):
     messages.append({"role": "user", "content": user_input})
     response = openai.ChatCompletion.create(
         model = "gpt-3.5-turbo",
@@ -21,10 +21,19 @@ def CustomChatGPT(user_input):
     )
     ChatGPT_reply = response["choices"][0]["message"]["content"]
     messages.append({"role": "assistant", "content": ChatGPT_reply})
-    return ChatGPT_reply
+    return ChatGPT_reply, messages
 
-demo = gradio.Interface(fn=CustomChatGPT, inputs = "text", outputs = "text", title = "Real Estate Video Script Writer")
+def wrapped_chat_gpt(user_input):
+    # Replace the following line with your method to retrieve the messages list for the current user
+    messages = initial_messages.copy()
 
-demo.launch(inline = False)
+    reply, updated_messages = CustomChatGPT(user_input, messages)
 
+    # Replace the following line with your method to store the updated messages list for the current user
+    # Store updated_messages
 
+    return reply
+
+demo = gradio.Interface(fn=wrapped_chat_gpt, inputs="text", outputs="text", title="Real Estate Video Script Writer")
+
+demo.launch(inline=False)