Update app.py

app.py

@@ -3,6 +3,7 @@ import openai
 import urllib.parse
 import io
 from PIL import Image
+import time  # Used for simulating progress updates
 
 # Function to call OpenAI GPT model for prompt processing
 def get_diagram_code(prompt, diagram_type, api_key):
@@ -162,19 +163,21 @@ def main():
     diagram_types = ["UML Diagram", "ER Diagram", "State Diagram", "Class Diagram", "Sequence Diagram"]
     diagram_choice = st.selectbox("Select the type of diagram to generate:", diagram_types)
 
-    # Show loading spinner when generating the diagram
-    loading_spinner = st.empty()
-    
+    # Create a progress bar
+    progress_bar = st.progress(0)
+
     if st.button("Generate Diagram"):
         if prompt:
-            # Show loading animation
-            loading_spinner.markdown("<div class='loading-spinner'></div>", unsafe_allow_html=True)
-            
+            # Simulate the progress
+            for i in range(100):
+                progress_bar.progress(i + 1)
+                time.sleep(0.05)  # Simulate processing time for loading bar
+
             diagram_code = get_diagram_code(prompt, diagram_choice, api_key)
             
             if diagram_code:
-                # Hide the spinner after diagram code is generated
-                loading_spinner.empty()
+                # Hide the progress bar once the diagram is generated
+                progress_bar.empty()
 
                 # Render Mermaid code to Streamlit with animation
                 st.code(diagram_code, language='mermaid')