fix chatbot loading and share link

app.py

@@ -1,16 +1,32 @@
 import gradio as gr
-from tabs.chat import create_chat_tab
+from tabs.chat import create_chat_tab, load_chatbot
 from tabs.create_chatbot import create_chatbot_tab
 from tabs.admin import create_admin_tab
 
 def create_app():
-    with gr.Blocks(title="codora AI App Creator v0") as demo:
+    with gr.Blocks(
+        title="codora AI App Creator v0",
+        head="""
+        <script>
+            function copyToClipboard() {
+                var copyText = document.querySelector("#share_link textarea").value;
+                navigator.clipboard.writeText(copyText).then(function() {
+                    alert("Copied the text: " + copyText);
+                }, function(err) {
+                    console.error("Could not copy text: ", err);
+                });
+            }
+        </script>
+        """
+    ) as demo:
         gr.Markdown("# codora AI App Creator v0")
-        chatbot_id_input = gr.Textbox(label="Enter Chatbot ID", visible=False)
+        chatbot_id_input = gr.Textbox(label="Enter Chatbot ID")
+        load_button = gr.Button("Load Chatbot")
+        load_message = gr.Textbox(label="Load Message", interactive=False)
         
         with gr.Tabs():
             with gr.Tab("Chat"):
-                chat_tab = create_chat_tab(chatbot_id_input)
+                chat_tab = create_chat_tab(chatbot_id_input, load_button, load_message)
             
             with gr.Tab("Create Chatbot"):
                 create_chatbot_tab()
@@ -18,28 +34,22 @@ def create_app():
             with gr.Tab("Admin"):
                 create_admin_tab()
 
-        @demo.load(inputs=[chatbot_id_input], outputs=[chat_tab['title'], chatbot_id_input, chat_tab['share_link']])
-        def load_chatbot(chatbot_id, request: gr.Request):
+        @demo.load(inputs=[chatbot_id_input], outputs=[chat_tab['title'], chatbot_id_input, chat_tab['share_link'], load_message])
+        def load_chatbot_on_start(chatbot_id, request: gr.Request):
             params = request.query_params
             if "chatbot_id" in params:
                 chatbot_id = params["chatbot_id"]
-
-            from database import get_chatbot
-            chatbot = get_chatbot(chatbot_id)
-            if chatbot:
-                return (
-                    gr.update(value=f"Welcome to {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}")
-                )
+                return load_chatbot(chatbot_id)
             return (
                 gr.update(value="Welcome to the Chatbot"),
                 gr.update(value="", visible=True),
+                gr.update(value=""),
                 gr.update(value="")
             )
 
     return demo
 
+demo = create_app()
+
 if __name__ == "__main__":
-    demo = create_app()
-    demo.launch()
+    demo.launch()

database.py

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
 
 try:
     logger.info("Connecting to database at %s", DB_HOST)
-    engine = create_engine(DATABASE_URL)
+    engine = create_engine(DATABASE_URL, connect_args={"connect_timeout": 10})
     # Test the connection
     with engine.connect() as connection:
         logger.info("Successfully connected to the database")
@@ -30,7 +30,6 @@ except Exception as e:
     logger.error(f"Error connecting to the database: {str(e)}")
     raise e
 
-#engine = create_engine(DATABASE_URL)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
 Base = declarative_base()
@@ -75,7 +74,7 @@ def generate_chatbot_id():
         'elephant', 'penguin', 'lion', 'dolphin', 'koala', 'panda', 'tiger', 'whale',
         'butterfly', 'eagle', 'peacock', 'unicorn', 'phoenix', 'otter', 'swan', 
         'chameleon', 'ladybug', 'flamingo', 'puppy', 'kitten', 'fawn', 'hummingbird', 
-        'koala', 'firefly', 'bunny', 'goldfish', 'puffin', 'orca', 'red panda', 'turtle', 
+        'koala', 'firefly', 'bunny', 'goldfish', 'puffin', 'orca', 'red_panda', 'turtle', 
         'parrot', 'owl', 'seahorse', 'hedgehog', 'sloth', 'duckling', 'starfish', 
         'gazelle', 'panther', 'robin', 'seal', 'lynx', 'jellyfish', 'gecko', 
         'kangaroo', 'lemur', 'meerkat', 'platypus', 'quokka', 'squirrel', 'toucan'
@@ -92,6 +91,9 @@ def create_chatbot(name, custom_instruction):
         db.commit()
         db.refresh(new_chatbot)
         return new_chatbot
+    except SQLAlchemyError as e:
+        db.rollback()
+        raise e
     finally:
         db.close()
 
@@ -99,6 +101,8 @@ def get_chatbot(chatbot_id):
     db = SessionLocal()
     try:
         return db.query(Chatbot).filter(Chatbot.chatbot_id == chatbot_id, Chatbot.is_active == True).first()
+    except SQLAlchemyError as e:
+        raise e
     finally:
         db.close()
 
@@ -116,6 +120,9 @@ def update_chatbot(chatbot_id, name=None, custom_instruction=None, is_active=Non
             db.commit()
             db.refresh(chatbot)
         return chatbot
+    except SQLAlchemyError as e:
+        db.rollback()
+        raise e
     finally:
         db.close()
 
@@ -128,6 +135,9 @@ def delete_chatbot(chatbot_id):
             db.commit()
             return True
         return False
+    except SQLAlchemyError as e:
+        db.rollback()
+        raise e
     finally:
         db.close()
 
@@ -135,5 +145,7 @@ def get_all_chatbots():
     db = SessionLocal()
     try:
         return db.query(Chatbot).all()
+    except SQLAlchemyError as e:
+        raise e
     finally:
         db.close()

tabs/chat.py

@@ -1,16 +1,48 @@
 import gradio as gr
-from utils.openai_utils import chat_with_bot
+from database import get_chatbot
 
-def create_chat_tab(chatbot_id_input):
+def load_chatbot(chatbot_id):
+    chatbot = get_chatbot(chatbot_id)
+    if chatbot:
+        return (
+            gr.update(value=f"Welcome to {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 loaded successfully.")
+        )
+    return (
+        gr.update(value="Welcome to the Chatbot"),
+        gr.update(value="", visible=True),
+        gr.update(value=""),
+        gr.update(value="Invalid chatbot ID or chatbot not active.")
+    )
+
+def create_chat_tab(chatbot_id_input, load_button, load_message):
     chatbot_title = gr.Markdown("Welcome to the Chatbot")
     chat_interface = gr.ChatInterface(
         chat_with_bot,
         additional_inputs=[chatbot_id_input],
     )
-    share_link = gr.Textbox(label="Share Link", interactive=False)
-    
+    share_link = gr.Textbox(label="Share Link", interactive=False, elem_id="share_link")
+
+    load_button.click(
+        fn=load_chatbot,
+        inputs=[chatbot_id_input],
+        outputs=[chatbot_title, chatbot_id_input, share_link, load_message]
+    )
+
+    # Add a custom HTML button for copying the share URL to the clipboard
+    copy_button = gr.HTML("""
+    <button onclick="copyToClipboard()">Copy to Clipboard</button>
+    """)
+
     return {
         "title": chatbot_title,
         "interface": chat_interface,
-        "share_link": share_link
-    }
+        "share_link": share_link,
+        "copy_button": copy_button
+    }
+
+def chat_with_bot(message, history, chatbot_id):
+    from utils.openai_utils import chat_with_bot as openai_chat_with_bot
+    return openai_chat_with_bot(message, history, chatbot_id)