Update app.py

app.py

@@ -30,6 +30,7 @@ import tempfile
 
 os.environ["OPENAI_API_KEY"] = "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"
 query1=" "
+limit = 0
 def loading_pdf():
     return "Loading..."
 
@@ -96,16 +97,19 @@ def infer(question):
     prompt_text = question
     if prompt_text:
         query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
-        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("'", "")
-    return result['answer']
+        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("'", "")
+            return result['answer']
+        else:
+            return "Usage Limit reached :("
 
 css="""
 #col-container { margin-left: auto; margin-right: auto;}