DB updated

database.py

@@ -165,11 +165,10 @@ def generate_chatbot_data(user_id, workspace_id, chatbot_id, chatbot_name, model
         },
         'mcp': mcp_config,
         'systemPrompt': system_prompt,
-        'label': '',
-        'iconPath': '',
         'errorText': '',
+        'replayType': 'multiple',
         'validationRegex': '',
-        'switchButtonEntities': [
+        'switchButtonModels': [
             {
                 'title': 'Back Button',
                 'value': True,
@@ -181,7 +180,7 @@ def generate_chatbot_data(user_id, workspace_id, chatbot_id, chatbot_name, model
                 'isDisabled': False
             }
         ],
-        'optionEntities': []
+        'optionModels': []
     }
 
     return {
@@ -194,12 +193,25 @@ def generate_chatbot_data(user_id, workspace_id, chatbot_id, chatbot_name, model
                 'value': 'asset/image/female_profile.png'
             },
             'branding': True,
+            'dynamicTheme': '{}',
+            'isDynamicTheme': False,
             'position': 'bottomRight',
             'primaryColor': 4287203839,
             'themeColor': 4294967295,
             'welcomeMessage': True
         },
         'id': chatbot_id,
+        "notificationModel": {
+            "sendTo": [
+                "afmjoaa@gmail.com",
+                "mprattoy@gmail.com"
+            ],
+            "subjectHeader": "New Response from Curious Learner",
+            "sendIncompleteResponse": True,
+            "leadInfo": "notificationCount",
+            "emailNotification": True
+        },
+        'responseCount': 0,
         'shareScript': generate_chatbot_share_script(user_id, workspace_id, chatbot_id)
     }
 
@@ -269,17 +281,17 @@ def create_agent(user_id, agent_name, model_name, mcp_github_enabled, github_tok
             # Path: g_users/{user_id}/workspaces/{DEFAULT_WORKSPACE_ID}/chatbots/{chatbot_id}
             chatbot_data['timestamp'] = firestore.SERVER_TIMESTAMP
             db.collection('g_users').document(user_id).collection('workspaces').document(
-                DEFAULT_WORKSPACE_ID).collection('chatbots').document(chatbot_id).set(chatbot_data)
+                DEFAULT_WORKSPACE_ID).collection('chats').document(chatbot_id).set(chatbot_data)
         else:
             mock_agents[agent_id] = agent_data
 
             # Store chatbot in mock workspace
             for user_email, user_data in mock_users.items():
                 if user_data['user_id'] == user_id:
-                    if 'chatbots' not in user_data['workspaces'][DEFAULT_WORKSPACE_ID]:
-                        user_data['workspaces'][DEFAULT_WORKSPACE_ID]['chatbots'] = {}
+                    if 'chats' not in user_data['workspaces'][DEFAULT_WORKSPACE_ID]:
+                        user_data['workspaces'][DEFAULT_WORKSPACE_ID]['chats'] = {}
                     chatbot_data['timestamp'] = datetime.now()
-                    user_data['workspaces'][DEFAULT_WORKSPACE_ID]['chatbots'][chatbot_id] = chatbot_data
+                    user_data['workspaces'][DEFAULT_WORKSPACE_ID]['chats'][chatbot_id] = chatbot_data
                     break
 
         success_msg = f"✅ Agent '{agent_name}' created successfully!"
@@ -302,13 +314,13 @@ def get_user_agents(user_id):
             # Fetch chatbots from the workspace subcollection
             # Path: g_users/{user_id}/workspaces/{DEFAULT_WORKSPACE_ID}/chatbots
             chatbots = db.collection('g_users').document(user_id).collection('workspaces').document(
-                DEFAULT_WORKSPACE_ID).collection('chatbots').get()
+                DEFAULT_WORKSPACE_ID).collection('chats').get()
             agent_list = []
             for chatbot in chatbots:
                 data = chatbot.to_dict()
                 chatbot_id = data.get('id', chatbot.id)
                 chatbot_name = data.get('chatbotName', 'Unnamed Agent')
-                agent_url = f"https://curious-learner-chatbot.web.app/#/chats?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
+                agent_url = f"https://curious-learner-chatbot.web.app/#/individual_chat_screen?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
                 agent_list.append(f"• {chatbot_name}\n  {agent_url}")
             return agent_list
         else:
@@ -317,10 +329,10 @@ def get_user_agents(user_id):
                 if user_data['user_id'] == user_id:
                     workspaces = user_data.get('workspaces', {})
                     default_workspace = workspaces.get(DEFAULT_WORKSPACE_ID, {})
-                    chatbots = default_workspace.get('chatbots', {})
+                    chatbots = default_workspace.get('chats', {})
                     for chatbot_id, data in chatbots.items():
                         chatbot_name = data.get('chatbotName', 'Unnamed Agent')
-                        agent_url = f"https://curious-learner-chatbot.web.app/#/chats?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
+                        agent_url = f"https://curious-learner-chatbot.web.app/#/individual_chat_screen?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
                         agent_list.append(f"• {chatbot_name}\n  {agent_url}")
                     break
             return agent_list

gradio_app.py

@@ -29,7 +29,7 @@ def get_chatbot_iframe(user_id):
     return f'''
     <iframe 
         src="{chatbot_url}" 
-        style="width: 100%; height: 100%; border: none;"
+        style="width: 100%; height: 600px; border: none;"
         allow="clipboard-write; clipboard-read"
     ></iframe>
     '''
@@ -93,13 +93,17 @@ def get_username(session_id):
 
 def create_agent(session_id, agent_name, model_name, mcp_github_enabled, github_token, system_prompt):
     """Create a new React agent"""
+    print(f"[DEBUG] create_agent: session_id={session_id}, agent_name={agent_name}")
     if not session_id or session_id not in user_sessions:
+        print(f"[DEBUG] create_agent: session not found")
         return "Please login first!", None
 
     user_id = user_sessions[session_id]
+    print(f"[DEBUG] create_agent: user_id={user_id}")
     success, message, chatbot_id = db_create_agent(
         user_id, agent_name, model_name, mcp_github_enabled, github_token, system_prompt
     )
+    print(f"[DEBUG] create_agent: success={success}, chatbot_id={chatbot_id}")
 
     return message, chatbot_id if success else None
 
@@ -232,8 +236,8 @@ custom_css = """
         position: fixed !important;
         bottom: 20px !important;
         right: 20px !important;
-        width: 380px !important;
-        height: 550px !important;
+        width: 400px !important;
+        height: 600px !important;
         border: none !important;
         border-radius: 16px !important;
         overflow: hidden !important;
@@ -244,7 +248,25 @@ custom_css = """
 
     #chatbot-webview iframe {
         width: 100% !important;
-        height: 100% !important;
+        height: 600px !important;
+        border: none !important;
+    }
+
+    /* Agent preview styling */
+    #agent-preview {
+        width: 100% !important;
+        min-height: 500px !important;
+        height: 500px !important;
+        border: 2px solid #e5e7eb;
+        border-radius: 12px;
+        overflow: hidden;
+        margin: 15px 0;
+        background: #f9fafb;
+    }
+
+    #agent-preview iframe {
+        width: 100% !important;
+        height: 500px !important;
         border: none !important;
     }
 
@@ -317,12 +339,6 @@ with gr.Blocks(title="React Agent Builder") as app:
                 # Logout button at the bottom of dashboard, inside the container
                 logout_btn_dashboard = gr.Button("🚪 Logout", size="sm", variant="secondary", elem_classes="logout-btn")
 
-            # Floating chatbot webview - outside the main container
-            chatbot_webview = gr.HTML(
-                value="",
-                elem_id="chatbot-webview"
-            )
-
         with gr.Tab("Create Agent", id="create_agent", visible=False) as create_agent_tab:
             with gr.Column(elem_classes="form-container"):
                 # Back button at the top
@@ -378,9 +394,20 @@ with gr.Blocks(title="React Agent Builder") as app:
                 create_msg = gr.Textbox(label="Status", interactive=False, lines=2, scale=1)
                 agent_url_output = gr.Textbox(label="Agent URL", interactive=False, lines=3, scale=1)
 
+                # Agent preview iframe - shows after successful creation
+                agent_preview_label = gr.Markdown("### 🤖 Agent Preview")
+                agent_preview = gr.HTML(value="", elem_id="agent-preview")
+
                 # Logout button at the bottom of create agent form
                 logout_btn_create = gr.Button("🚪 Logout", size="sm", variant="secondary", elem_classes="logout-btn")
 
+    # Floating chatbot webview - outside tabs so it's always available
+    chatbot_webview = gr.HTML(
+        value="",
+        elem_id="chatbot-webview",
+        visible=False
+    )
+
 
     # Event handlers
     def handle_login(email, password):
@@ -408,7 +435,7 @@ with gr.Blocks(title="React Agent Builder") as app:
                 session_state: sess_id,
                 user_id_state: user_id,
                 agents_list: agents_text,
-                chatbot_webview: chatbot_html,
+                chatbot_webview: gr.update(value=chatbot_html, visible=True),
                 tabs: gr.update(selected="dashboard")
             }
         else:
@@ -439,7 +466,7 @@ with gr.Blocks(title="React Agent Builder") as app:
             login_email: "",
             login_password: "",
             login_msg: "",
-            chatbot_webview: "",
+            chatbot_webview: gr.update(value="", visible=False),
             tabs: gr.update(selected="login")
         }
 
@@ -448,11 +475,13 @@ with gr.Blocks(title="React Agent Builder") as app:
         return {
             dashboard_tab: gr.update(visible=False),
             create_agent_tab: gr.update(visible=True),
+            chatbot_webview: gr.update(visible=False),
             tabs: gr.update(selected="create_agent")
         }
 
 
-    def switch_back_to_dashboard():
+    def switch_back_to_dashboard(user_id):
+        chatbot_html = get_chatbot_iframe(user_id) if user_id else ""
         return {
             dashboard_tab: gr.update(visible=True),
             create_agent_tab: gr.update(visible=False),
@@ -462,6 +491,8 @@ with gr.Blocks(title="React Agent Builder") as app:
             system_prompt_input: "",
             create_msg: "",
             agent_url_output: "",
+            agent_preview: "",
+            chatbot_webview: gr.update(value=chatbot_html, visible=True),
             tabs: gr.update(selected="dashboard")
         }
 
@@ -472,22 +503,26 @@ with gr.Blocks(title="React Agent Builder") as app:
 
 
     def handle_agent_creation(sess_id, user_id, name, model, mcp_enabled, github_token, prompt):
+        print(f"[DEBUG] handle_agent_creation: sess_id={sess_id}, user_id={user_id}, name={name}")
         msg, chatbot_id = create_agent(sess_id, name, model, mcp_enabled, github_token, prompt)
+        print(f"[DEBUG] handle_agent_creation: msg={msg}, chatbot_id={chatbot_id}")
 
-        # Build the agent URL locally if creation was successful
+        # Build the individual agent URL locally if creation was successful
         agent_url = ""
+        agent_iframe = ""
         if chatbot_id and user_id:
-            agent_url = f"https://curious-learner-chatbot.web.app/#/chats?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
+            agent_url = f"https://curious-learner-chatbot.web.app/#/individual_chat_screen?clientId={user_id}&workspaceId={DEFAULT_WORKSPACE_ID}&chatbotId={chatbot_id}"
+            print(f"[DEBUG] handle_agent_creation: agent_url={agent_url}")
+            agent_iframe = f'''<iframe src="{agent_url}" style="width: 100%; height: 500px; border: none;" allow="clipboard-write; clipboard-read"></iframe>'''
+            print(f"[DEBUG] handle_agent_creation: iframe built, length={len(agent_iframe)}")
+        else:
+            print(f"[DEBUG] handle_agent_creation: chatbot_id or user_id is None/empty")
 
-        # Also refresh the agents list and webview after creation
+        # Also refresh the agents list after creation
         agent_list = db_get_user_agents(user_id) if user_id else []
         updated_agents = "\n\n".join(agent_list) if agent_list else "No agents created yet."
-        return {
-            create_msg: msg,
-            agent_url_output: agent_url,
-            agents_list: updated_agents,
-            chatbot_webview: get_chatbot_iframe(user_id)
-        }
+
+        return msg, agent_url, agent_iframe, updated_agents
 
 
     # Refresh agents list function
@@ -499,7 +534,16 @@ with gr.Blocks(title="React Agent Builder") as app:
         return "\n\n".join(agent_list) if agent_list else "No agents created yet."
 
 
-    # Login event - chain refresh after login
+    # Load chatbot webview function
+    def load_chatbot_webview(user_id):
+        """Load chatbot webview for user"""
+        if not user_id:
+            return gr.update(value="", visible=False)
+        chatbot_html = get_chatbot_iframe(user_id)
+        return gr.update(value=chatbot_html, visible=True)
+
+
+    # Login event - chain refresh and webview load after login
     login_btn.click(
         fn=handle_login,
         inputs=[login_email, login_password],
@@ -509,6 +553,10 @@ with gr.Blocks(title="React Agent Builder") as app:
         fn=refresh_agents_list,
         inputs=[user_id_state],
         outputs=[agents_list]
+    ).then(
+        fn=load_chatbot_webview,
+        inputs=[user_id_state],
+        outputs=[chatbot_webview]
     )
 
     # Logout events (both buttons do the same thing)
@@ -535,14 +583,16 @@ with gr.Blocks(title="React Agent Builder") as app:
     # Create agent button
     create_agent_btn.click(
         fn=switch_to_create_agent,
-        outputs=[dashboard_tab, create_agent_tab, tabs]
+        outputs=[dashboard_tab, create_agent_tab, chatbot_webview, tabs]
     )
 
     # Back button
     back_btn.click(
         fn=switch_back_to_dashboard,
+        inputs=[user_id_state],
         outputs=[dashboard_tab, create_agent_tab, agent_name_input, mcp_github,
-                 github_token_input, system_prompt_input, create_msg, agent_url_output, tabs]
+                 github_token_input, system_prompt_input, create_msg, agent_url_output, agent_preview, chatbot_webview,
+                 tabs]
     )
 
     # MCP GitHub checkbox toggle
@@ -557,7 +607,7 @@ with gr.Blocks(title="React Agent Builder") as app:
         fn=handle_agent_creation,
         inputs=[session_state, user_id_state, agent_name_input, model_dropdown, mcp_github,
                 github_token_input, system_prompt_input],
-        outputs=[create_msg, agent_url_output, agents_list, chatbot_webview]
+        outputs=[create_msg, agent_url_output, agent_preview, agents_list]
     )
 
     # Refresh agents list button