Update app.py

app.py

@@ -1,26 +1,30 @@
 import streamlit as st
 import openai
 import os
+import time
 
 openai.api_key = os.environ.get("OPEN_AI_API")
 st.title("SEO Article Writer with ChatGPT")
 
+
 def generate_article(keyword, writing_style, word_count):
-    #return "This is a test article generated without making API calls."
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
                 {"role": "user", "content": "Write a SEO optimized word article about " + keyword},
                 {"role": "user", "content": "The article should be " + writing_style},
                 {"role": "user", "content": "The article length should " + str(word_count)},
             ]
-    )
-    result = ''
-    for choice in response.choices:
-        result += choice.message.content
-
-    print(result)
-    return result
+        )
+        result = ''
+        for choice in response.choices:
+            result += choice.message.content
+        return result
+    except openai.error.RateLimitError:
+        st.error("Rate limit exceeded. Waiting for 20 seconds before retrying...")
+        time.sleep(20)
+        return generate_article(keyword, writing_style, word_count) 
 
 keyword = st.text_input("Enter a keyword:")
 writing_style = st.selectbox("Select writing style:", ["Casual", "Informative", "Professional"])