Updated app.py

app.py

@@ -1,10 +1,32 @@
-## Conversational Q&A Chatbot
 import streamlit as st
+import time
+from langchain.schema import HumanMessage, SystemMessage, AIMessage
 
-from langchain.schema import HumanMessage,SystemMessage,AIMessage
-from langchain.chat_models import ChatOpenAI
-
-## Streamlit UI
+def get_chatmodel_response(question):
+    # Retry logic
+    max_retries = 3
+    retries = 0
+
+    while retries < max_retries:
+        try:
+            st.session_state['flowmessages'].append(HumanMessage(content=question))
+            answer = chat(st.session_state['flowmessages'])
+            st.session_state['flowmessages'].append(AIMessage(content=answer.content))
+            return answer.content
+        except Exception as e:
+            print(f"Error: {e}")
+            if "Rate limit" in str(e):
+                print(f"Rate limit exceeded. Waiting and retrying...")
+                time.sleep(5)  # Adjust the waiting time as needed
+                retries += 1
+            else:
+                print("Unhandled exception. Please try again later.")
+                break
+
+    print("Exceeded the maximum number of retries. Please try again later.")
+    return None
+
+# Streamlit app setup
 st.set_page_config(page_title="Conversational Q&A Chatbot")
 st.header("Hey, Let's Chat")
 
@@ -12,29 +34,24 @@ from dotenv import load_dotenv
 load_dotenv()
 import os
 
-chat=ChatOpenAI(temperature=0.5)
+# Assuming the existence of the ChatOpenAI class
+chat = ChatOpenAI(temperature=0.5)
 
 if 'flowmessages' not in st.session_state:
-    st.session_state['flowmessages']=[
-        SystemMessage(content="Yor are a comedian AI assitant")
+    st.session_state['flowmessages'] = [
+        SystemMessage(content="You are a comedian AI assistant")
     ]
 
-## Function to load OpenAI model and get respones
-
-def get_chatmodel_response(question):
-
-    st.session_state['flowmessages'].append(HumanMessage(content=question))
-    answer=chat(st.session_state['flowmessages'])
-    st.session_state['flowmessages'].append(AIMessage(content=answer.content))
-    return answer.content
-
-input=st.text_input("Input: ",key="input")
-response=get_chatmodel_response(input)
-
-submit=st.button("Ask the question")
-
-## If ask button is clicked
+# Streamlit UI
+input_question = st.text_input("Input: ", key="input")
+submit = st.button("Ask the question")
 
+# If the "Ask" button is clicked
 if submit:
-    st.subheader("The Response is")
-    st.write(response)
+    response = get_chatmodel_response(input_question)
+
+    if response is not None:
+        st.subheader("The Response is:")
+        st.write(response)
+    else:
+        st.subheader("Error: Unable to get response. Please try again later.")