'u

app.py

@@ -23,11 +23,12 @@ def login_user(username, password):
         return True
     print('login faild')
     print(users_db)
+    st.session_state['logged_in'] = 'login'
     return False
 
 def logout_user():
     """Log out the current user."""
-    st.session_state['logged_in'] = False
+    st.session_state['logged_in'] = 'login'
     st.session_state.pop('user', None)
     st.session_state.pop('step1_done', None)
     st.session_state.pop('step2_done', None)
@@ -59,45 +60,51 @@ def show_login():
             st.error("Invalid username or password.")
 
 def show_task():
-    """Display the multi-step task."""
-    st.title(f"Welcome, {st.session_state['user']['name']}!")
-
-    # Step 1: Confirm personal details
-    st.header("Step 1: Confirm Your Details")
-    name = st.text_input("Name", value=st.session_state['user']['name'])
-    age = st.number_input("Age", value=st.session_state['user']['age'], min_value=0, max_value=120, step=1)
-    if st.button("Submit Step 1"):
-        st.session_state['user']['name'] = name
-        st.session_state['user']['age'] = age
-        st.session_state['step1_done'] = True
-        st.write(f"Hello, {name}. You are {age} years old.")
-        time.sleep(1)  # Simulate a processing delay
-
-    # Step 2: Update favorite color
-    if 'step1_done' in st.session_state:
-        st.header("Step 2: Update Your Favorite Color")
-        favorite_color = st.text_input("What's your favorite color?", value=st.session_state['user']['favorite_color'])
-        if st.button("Submit Step 2"):
-            st.session_state['user']['favorite_color'] = favorite_color
-            st.session_state['step2_done'] = True
-            st.write(f"Got it! Your favorite color is {favorite_color}.")
-            time.sleep(1)  # Simulate a processing delay
-
-    # Step 3: Final Confirmation and Real-time Processing
-    if 'step2_done' in st.session_state:
-        st.header("Step 3: Final Confirmation")
-        confirm = st.checkbox("Confirm and process the data")
-        if confirm:
-            st.write("Processing data...")
-            progress_bar = st.progress(0)
-
-            # Simulate a real-time processing task
-            for i in range(100):
-                progress_bar.progress(i + 1)
-                time.sleep(0.05)  # Simulate real-time data streaming
-
-            st.success("Task Completed!")
-            st.write(f"Summary: {name}, {age} years old, likes {favorite_color}.")
+    
+    st.title("Task")
+
+    # init session_state
+    if 'step1' not in st.session_state:
+        st.session_state['step1'] = None
+    if 'step2' not in st.session_state:
+        st.session_state['step2'] = None
+    if 'step3' not in st.session_state:
+        st.session_state['step3'] = None
+    if 'step4' not in st.session_state:
+        st.session_state['step4'] = None    
+
+    st.header('step 1: 文案准备')
+    st.text_input("请输入文案关键词", key="step1_text")
+    st.checkbox("风格", key="step1_voice")
+    if st.button("开始生成文案"):        
+        st.session_state['step1'] = 'done'
+    
+    if st.session_state['step1'] == 'done':
+        st.header('step 2: 语音风格')
+        st.text_input("语音风格关键词", key="step2_text")
+        if st.button("开始生成语音"):
+            st.session_state['step2'] = 'done'
+    
+    if st.session_state['step2'] == 'done':
+        st.header('step 3: 素材风格')
+        st.text_input("请输入素材风格关键词", key="step3_text")
+        if st.button("开始载入素材"):
+            st.session_state['step3'] = 'done'
+    
+    if st.session_state['step3'] == 'done':
+        st.header('step 4: 视频风格')
+        st.text_input("请输入视频风格关键词", key="step4_text")
+        if st.button("开始合并视频"):
+            st.session_state['step4'] = 'done'
+    
+    if st.session_state['step4'] == 'done':
+        progress_bar = st.progress(0)
+
+        # Simulate a real-time processing task
+        for i in range(100):
+            progress_bar.progress(i + 1)
+            time.sleep(0.05)  # Simulate real-time data streaming
+
 
     # Option to logout
     if st.button("Logout"):
@@ -108,12 +115,13 @@ def main():
     st.title("Streamlit Application with Login and Multi-step Task")
 
     if 'logged_in' not in st.session_state:
-        st.session_state['logged_in'] = False
+        st.session_state['logged_in'] = 'login'
 
-    if not st.session_state['logged_in']:
-        show_registration()
+    if st.session_state['logged_in'] == 'login':
         show_login()
-    else:
+    elif st.session_state['logged_in'] == 'register':
+        show_registration()
+    elif st.session_state['logged_in'] == True:
         show_task()
 
 if __name__ == "__main__":