Update app.py

app.py

@@ -14,7 +14,7 @@ def generate_conversation_id():
 # Handle new chat creation
 def new_chat():
     conversation_id = generate_conversation_id()
-    return f"New conversation created. Share this URL to join: /?chat={conversation_id}"
+    return conversation_id
 
 # Handle user joining the conversation
 def join_chat(chat_id, username):
@@ -41,26 +41,43 @@ def chat_interface(chat_id, username, message):
 # Gradio app
 def main():
     with gr.Blocks() as demo:
-        gr.Markdown("# Chat Application\nClick 'New Chat' to start a new conversation or join an existing one.")
-        new_chat_button = gr.Button("New Chat")
-        new_chat_output = gr.Textbox(label="New Chat URL")
+        gr.Markdown("# Chat Application\nChoose an option to start or join a conversation.")
         
-        new_chat_button.click(fn=new_chat, outputs=new_chat_output)
+        # Initial buttons for New Chat or Enter Conversation ID
+        with gr.Column() as initial_buttons:
+            new_chat_button = gr.Button("New Chat")
+            existing_chat_button = gr.Button("Enter Conversation ID for Existing Chat")
         
-        with gr.Row():
+        # Chat interface components, initially hidden
+        with gr.Column(visible=False) as chat_interface:
+            chat_id_display = gr.Markdown("Chat ID: ")
             chat_id = gr.Textbox(label="Chat ID", placeholder="Enter Chat ID to join or create a new one.")
             username = gr.Textbox(label="Username", placeholder="Enter your username.")
             join_button = gr.Button("Join Chat")
             join_output = gr.Textbox(label="Join Status")
-            
-            join_button.click(fn=join_chat, inputs=[chat_id, username], outputs=join_output)
+            message = gr.Textbox(label="Your Message", placeholder="Type your message here.")
+            send_button = gr.Button("Send Message")
+            chat_output = gr.Textbox(label="Chat Messages")
         
-        message = gr.Textbox(label="Your Message", placeholder="Type your message here.")
-        send_button = gr.Button("Send Message")
-        chat_output = gr.Textbox(label="Chat Messages")
+        # New Chat button logic
+        def handle_new_chat():
+            conversation_id = new_chat()
+            return gr.update(visible=True), gr.update(value=f"Chat ID: {conversation_id}"), conversation_id
         
+        new_chat_button.click(fn=handle_new_chat, outputs=[chat_interface, chat_id_display, chat_id])
+        
+        # Existing Chat button logic
+        def handle_existing_chat():
+            return gr.update(visible=True)
+        
+        existing_chat_button.click(fn=handle_existing_chat, outputs=chat_interface)
+        
+        # Join Chat button logic
+        join_button.click(fn=join_chat, inputs=[chat_id, username], outputs=join_output)
+        
+        # Send Message button logic
         send_button.click(fn=chat_interface, inputs=[chat_id, username, message], outputs=chat_output)
-
+    
     demo.launch()
 
 if __name__ == "__main__":