From: ibidyouadu <angel.d.umana@gmail.com>
Date: Sun, 20 Aug 2023 00:55:06 +0000 (-0500)
Subject: update .gitignore and add redis client
X-Git-Url: http://git.angelumana.com/?a=commitdiff_plain;h=cef21c0e8d4f5a30a267868385b8a45c60bdf149;p=diane%2F.git

update .gitignore and add redis client
---

diff --git a/.gitignore b/.gitignore
index 0bbae6f..0be4a4a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 .DS_Store
 __pycache__/
 .env
+*.rdb
\ No newline at end of file
diff --git a/main.py b/main.py
index 0f04567..34866db 100644
--- a/main.py
+++ b/main.py
@@ -4,37 +4,66 @@ from twilio.twiml.messaging_response import MessagingResponse
 import random
 import os
 import openai
+import redis
 from settings import OPENAI_API_KEY, OPENWEATHER_API_KEY
 
 openai.api_key = OPENAI_API_KEY
 
 app = Flask(__name__)
 
+# Initiate redis connection
+r = redis.Redis(host="localhost", port=6379, decode_responses=True)
+
+# Wipe previous conversation data and start a new one with assistant prompt
+conversation_key = "main"
+r.delete(conversation_key)
+prompt = {
+    "role": "system",
+    "content": "You are a helpful assistant named Diane."
+}
+r.rpush(conversation_key, prompt)
+
 @app.route("/", methods=["POST"])
 
 def main():
-    input_msg = request.values.get("Body", "")
+    # Get user input
+    input_text = request.values.get("Body", "")
+
+    # Add to payload for OpenAI API
+    input_message = {
+        "role": "user",
+        "content": input_text
+    }
+
+    # Twilio API. We will put text content in `msg` attributes and return a string representation of `response``
     response = MessagingResponse()
     msg = response.message()
 
+    # Retrieve conversation from redis in format ready to post to OpenAI. Update redis db with new input
+    conversation = r.lrange(conversation_key, 0, -1)
+    messages = [eval(message) for message in conversation]
+    messages.append(input_message)
+    print(messages)
+    r.rpush(conversation_key, input_message)
+
+    # Call OpenAI API
     openai_res = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        messages=[
-            {
-                "role": "system",
-                "content": "You are a helpful assistant named Diane."
-            },
-            {
-                "role": "user",
-                "content": input_msg
-            }
-        ],
+        messages=messages,
         max_tokens=256,
         temperature=0.6
     )
 
-    msg.body(openai_res.choices[0].message.content)
+    # Update redis db with OpenAI response
+    output_text = openai_res.choices[0].message.content
+    output_message = {
+        "role": "assistant",
+        "content": output_text
+    }
+    r.rpush(conversation_key, output_message)
 
+    # Return OpenAI message
+    msg.body(output_text)
     return str(response)
 
 if __name__ == "__main__":