Update app.py

app.py

@@ -54,17 +54,21 @@ with gr.Blocks() as demo:
     clear = gr.Button("Clear")
 
     def respond(openai_api_key, message, chat_history):
-        openai.api_key = openai_api_key
-        index = load_index_from_storage(storage_context)
-        query_engine = index.as_query_engine(similarity_top_k=3)
-
-        response = query_engine.query(message)
-        bot_message = response.response
-        for i, row in get_podcast_and_youtube(response).iterrows():
-            bot_message += f"\n\n\n Source: {row.podcast} \n\n Link: {row.youtube_id}"
-        chat_history.append((message, bot_message))
-        time.sleep(1)
-        return "", chat_history
+        if not openai_api_key:
+            return "No OpenAI key provided, please provide one.", chat_history
+        try:
+            openai.api_key = openai_api_key
+            index = load_index_from_storage(storage_context)
+            query_engine = index.as_query_engine(similarity_top_k=3)
+            response = query_engine.query(message)
+            bot_message = response.response
+            for i, row in get_podcast_and_youtube(response).iterrows():
+                bot_message += f"\n\n\n Source: {row.podcast} \n\n Link: {row.youtube_id}"
+            chat_history.append((message, bot_message))
+            time.sleep(1)
+            return "", chat_history
+        except Exception as e:
+            return f"Unexpected error raised: {e}", chat_history
 
     msg.submit(respond, [openai_api_key, msg, chatbot], [msg, chatbot])
     btn2.click(respond, [openai_api_key, msg, chatbot], [msg, chatbot])