Update app.py

app.py

@@ -65,12 +65,11 @@ layer_agent_config_rec = {
 def stream_response(messages: Iterable[ResponseChunk]):
     layer_outputs = {}  # Store outputs per layer
     progress_bar = st.progress(0)  # Initialize progress bar
-    total_steps = len(messages)  # Estimate total messages for progress tracking
-    current_step = 0
+    current_step = 0  # Start step counter
 
     for message in messages:
         current_step += 1
-        progress_bar.progress(current_step / total_steps)  # Update progress bar
+        progress_bar.progress(current_step % 100)  # Use a modular percentage for progress
 
         if message['response_type'] == 'intermediate':
             layer = message['metadata']['layer']  # Identify the layer
@@ -79,22 +78,22 @@ def stream_response(messages: Iterable[ResponseChunk]):
             layer_outputs[layer].append(message['delta'])  # Add intermediate delta
             
             # Render intermediate output in real-time
-            with st.container():  # Create a dynamic container for rendering
+            with st.container():
                 st.markdown(f"### Layer {layer} (In Progress)")
                 for output in layer_outputs[layer]:
-                    st.markdown(f"- {output}")  # Render each intermediate result
-
+                    st.markdown(f"- {output}")
         else:
-            # Final rendering of outputs after processing completes for this layer
+            # Final rendering of outputs
             for layer, outputs in layer_outputs.items():
                 st.markdown(f"### Layer {layer} Final Output")
                 for output in outputs:
                     st.markdown(f"- {output}")
-            
-            layer_outputs = {}  # Reset layer outputs for the next iteration
-            yield message['delta']  # Yield the final processed result
 
-    progress_bar.empty()  # Clear progress bar after all processing is complete
+            layer_outputs = {}  # Reset layer outputs for next iteration
+            yield message['delta']  # Yield final processed result
+
+    progress_bar.empty()  # Clear progress bar after all processing
+
 
 
 def set_moa_agent(