add chatbot loading logging and timeout

tabs/chat.py

@@ -1,21 +1,62 @@
 import gradio as gr
+import logging
 from database import get_chatbot
+import concurrent.futures
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 def load_chatbot(chatbot_id):
-    chatbot = get_chatbot(chatbot_id)
-    if chatbot:
+    logger.info(f"Attempting to load chatbot with ID: {chatbot_id}")
+    try:
+        chatbot = get_chatbot(chatbot_id)
+        if chatbot:
+            logger.info(f"Chatbot {chatbot_id} loaded successfully")
+            return (
+                gr.update(value=f"# {chatbot.name}\n\nWillkommen bei {chatbot.name}"),
+                gr.update(value=chatbot_id, visible=True),
+                gr.update(value=f"https://huggingface.co/spaces/codora/ai-app-creator?chatbot_id={chatbot_id}"),
+                gr.update(value="Chatbot erfolgreich geladen.")
+            )
+        logger.warning(f"Chatbot {chatbot_id} not found or inactive")
         return (
-            gr.update(value=f"# {chatbot.name}\n\nWillkommen bei {chatbot.name}"),
-            gr.update(value=chatbot_id, visible=True),
-            gr.update(value=f"https://huggingface.co/spaces/codora/ai-app-creator?chatbot_id={chatbot_id}"),
-            gr.update(value="Chatbot erfolgreich geladen.")
+            gr.update(value="# Chatbot\n\nWillkommen beim Chatbot"),
+            gr.update(value="", visible=True),
+            gr.update(value=""),
+            gr.update(value="Ungültige Chatbot-ID oder Chatbot nicht aktiv.")
         )
-    return (
-        gr.update(value="# Chatbot\n\nWillkommen beim Chatbot"),
-        gr.update(value="", visible=True),
-        gr.update(value=""),
-        gr.update(value="Ungültige Chatbot-ID oder Chatbot nicht aktiv.")
-    )
+    except Exception as e:
+        logger.error(f"Error loading chatbot {chatbot_id}: {str(e)}")
+        return (
+            gr.update(value="# Chatbot\n\nWillkommen beim Chatbot"),
+            gr.update(value="", visible=True),
+            gr.update(value=""),
+            gr.update(value=f"Ein Fehler ist aufgetreten: {str(e)}")
+        )
+
+def load_chatbot_with_timeout(chatbot_id, timeout=5):
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        future = executor.submit(load_chatbot, chatbot_id)
+        try:
+            result = future.result(timeout=timeout)
+            return result
+        except concurrent.futures.TimeoutError:
+            logger.error(f"Timeout while loading chatbot {chatbot_id}")
+            return (
+                gr.update(value="# Chatbot\n\nWillkommen beim Chatbot"),
+                gr.update(value="", visible=True),
+                gr.update(value=""),
+                gr.update(value="Zeitüberschreitung beim Laden des Chatbots.")
+            )
+        except Exception as e:
+            logger.error(f"Unexpected error while loading chatbot {chatbot_id}: {str(e)}")
+            return (
+                gr.update(value="# Chatbot\n\nWillkommen beim Chatbot"),
+                gr.update(value="", visible=True),
+                gr.update(value=""),
+                gr.update(value=f"Ein unerwarteter Fehler ist aufgetreten: {str(e)}")
+            )
 
 def create_chat_tab():
     chatbot_id_input = gr.Textbox(label="Chatbot-ID eingeben")
@@ -38,7 +79,7 @@ def create_chat_tab():
     share_link = gr.Textbox(label="Link zum Teilen", interactive=False, elem_id="share_link")
 
     load_button.click(
-        fn=load_chatbot,
+        fn=load_chatbot_with_timeout,
         inputs=[chatbot_id_input],
         outputs=[chatbot_title, chatbot_id_input, share_link, load_message]
     )