Update app.py

app.py

@@ -27,10 +27,15 @@ from langchain.llms import OpenAI
 from langchain.document_loaders import PyPDFLoader
 from langchain.indexes import VectorstoreIndexCreator
 import tempfile
+import time
 
 os.environ["OPENAI_API_KEY"] = os.environ['OpenApi_Key']
 query1=" "
 limit = 0
+st = 0
+paused = False
+waittime = 10.0
+maxLimit = 5
 def loading_pdf():
     return "Loading..."
 
@@ -95,24 +100,36 @@ def bot(history):
 def infer(question):
     global query1
     global limit
+    global st
+    global paused
     openai.api_key = os.environ['OpenApi_Key']
     prompt_text = question
-    if prompt_text:
-        query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
-        if limit <= 5:
-            result = chain(query1)
-            query1 = query1 + result['answer']
-            query1 = openai.ChatCompletion.create(
-                                model="gpt-3.5-turbo",
-                                messages=[
-                                {"role": "system", "content": "You are provided with chat history and latset conversation between user and bot. Summarise the history and latest conversationin minimum most tokens possible. Do not include greetings in the summary like hi, hello, etc."},
-                                {"role": "user", "content": query1},
-                                    ]
-                                    ) ["choices"][0]["message"]["content"].replace("'", "")
-            limit += 1
-            return result['answer']
-        else:
-            return "Usage Limit reached :("
+    if paused == False:
+      if prompt_text:
+          query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
+          if limit <= maxLimit:
+              result = chain(query1)
+              query1 = query1 + result['answer']
+              query1 = openai.ChatCompletion.create(
+                                  model="gpt-3.5-turbo",
+                                  messages=[
+                                  {"role": "system", "content": "You are provided with chat history and latset conversation between user and bot. Summarise the history and latest conversationin minimum most tokens possible. Do not include greetings in the summary like hi, hello, etc."},
+                                  {"role": "user", "content": query1},
+                                      ]
+                                      ) ["choices"][0]["message"]["content"].replace("'", "")
+              limit += 1
+              if limit == 2:
+                paused = True
+                st = time.time()
+              return result['answer']
+    else:
+      if time.time() - st > waittime:
+        paused = False
+        limit = 0
+        return infer(question)
+      else:
+        return "Usage Limit reached :( Please visit https://edith.framer.ai/pricing to unlock unlimited access!"
+
 
 css="""
 #col-container { margin-left: auto; margin-right: auto;}