fixing the none output issue

app.py

@@ -36,33 +36,80 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"], templa
 
 OPENAI_API_KEY = ''
 
-def slow_echo(usr_message, chat_history):
-    global OPENAI_API_KEY
+# def slow_echo(usr_message, chat_history):
+#     global OPENAI_API_KEY
     
-    # Check if the API key is set
-    if not OPENAI_API_KEY:
-        error_message = "OpenAI API key not set. Please provide the key first."
-        print(error_message)
-        return error_message, chat_history
+#     # Check if the API key is set
+#     if not OPENAI_API_KEY:
+#         error_message = "OpenAI API key not set. Please provide the key first."
+#         print(error_message)
+#         return error_message, chat_history
     
-    try:
-        chat_model = ChatOpenAI(temperature=0, model_name='gpt-3.5-turbo', openai_api_key=OPENAI_API_KEY)
-
-        # customized memory
-        memory = ConversationBufferMemory(
-            return_messages=True,
-            output_key='result'
-        )
+#     try:
+#         chat_model = ChatOpenAI(temperature=0, model_name='gpt-3.5-turbo', openai_api_key=OPENAI_API_KEY)
+
+#         # customized memory
+#         memory = ConversationBufferMemory(
+#             return_messages=True,
+#             output_key='result'
+#         )
+        
+#         answer_chain = RetrievalQA.from_chain_type(
+#             chat_model,
+#             retriever=VectorStore.as_retriever(search_type="similarity", k=6),
+#             memory = memory,
+#             chain_type_kwargs={"prompt": QA_CHAIN_PROMPT},
+#             return_source_documents=True
+#         )
         
-        answer_chain = RetrievalQA.from_chain_type(
-            chat_model,
-            retriever=VectorStore.as_retriever(search_type="similarity", k=6),
-            memory = memory,
-            chain_type_kwargs={"prompt": QA_CHAIN_PROMPT},
-            return_source_documents=True
-        )
+#         # Get a response from the OpenAI model
+#         bot_result = answer_chain({"query": usr_message})
+#         bot_response = bot_result['result']
+#         source_doc = [bot_result['source_documents'][i].metadata['title'] for i in range(len(bot_result))]
+#         source_page = [str(bot_result['source_documents'][i].metadata['page']+1) for i in range(len(bot_result))]
+
+#         # formated output
+#         source_print = {}
+#         for i in range(len(source_doc)):
+#           if source_doc[i] in source_print:
+#             source_print[source_doc[i]] = source_print[source_doc[i]] + ', ' + source_page[i]
+#           else:
+#             source_print[source_doc[i]] = 'page: '+ source_page[i]
+
+#         bot_response += '\n Source:'
+#         for doc, page in source_print.items():
+#             bot_response += '\n' + doc + ': ' + page
+
+#         chat_history.append((usr_message, bot_response))
         
-        # Get a response from the OpenAI model
+#         time.sleep(1)
+#         yield "", chat_history
+
+#     except openai.error.OpenAIError as e:
+#         # Handle OpenAI-specific errors
+#         error_message = f"An openAI API Error: {e}"
+#         return error_message, chat_history
+
+#     except Exception as e:
+#         # Handle other unexpected errors
+#         error_message = f"An unexpected error: {e}"
+#         return error_message, chat_history
+
+# def get_opeanai_key(openai_key):
+#     global OPENAI_API_KEY
+#     OPENAI_API_KEY=openai_key
+
+#     return {chatbot_col: gr.Column(visible=True)}
+
+
+def slow_echo(usr_message, chat_history):
+    global request_count
+    try:
+      # Check if request count has reached the limit
+        if request_count >= 10:
+            return "You have reached the maximum number of allowed requests. Please exit the chat."
+
+        # Attempt to get a response from the OpenAI model
         bot_result = answer_chain({"query": usr_message})
         bot_response = bot_result['result']
         source_doc = [bot_result['source_documents'][i].metadata['title'] for i in range(len(bot_result))]
@@ -76,30 +123,26 @@ def slow_echo(usr_message, chat_history):
           else:
             source_print[source_doc[i]] = 'page: '+ source_page[i]
 
-        bot_response += '\n Source:'
+        bot_response = bot_response + '\n Source:'
         for doc, page in source_print.items():
             bot_response += '\n' + doc + ': ' + page
 
         chat_history.append((usr_message, bot_response))
-        
+
+
         time.sleep(1)
+
         yield "", chat_history
 
     except openai.error.OpenAIError as e:
         # Handle OpenAI-specific errors
-        error_message = f"An openAI API Error: {e}"
-        return error_message, chat_history
+        error_message = f"OpenAI API Error: {e}"
+        print(error_message)
 
     except Exception as e:
         # Handle other unexpected errors
-        error_message = f"An unexpected error: {e}"
-        return error_message, chat_history
-
-def get_opeanai_key(openai_key):
-    global OPENAI_API_KEY
-    OPENAI_API_KEY=openai_key
-
-    return {chatbot_col: gr.Column(visible=True)}
+        error_message = f"Unexpected error: {e}"
+        print(error_message)
 
 
 with gr.Blocks() as demo: