Update app.py

app.py

@@ -9,6 +9,7 @@ import io
 import time
 from streamlit.components.v1 import html
 import asyncio
+import base64
 
 if not asyncio.get_event_loop().is_running():
     asyncio.set_event_loop(asyncio.new_event_loop())
@@ -21,17 +22,21 @@ if 'processed_data' not in st.session_state:
         'audio': None
     }
 
+if 'image_data' not in st.session_state:
+    st.session_state.image_data = None
 
-# JavaScript timer component
+# JavaScript timer component with stop function
 def timer():
     return """
     <div id="timer" style="font-size:16px;color:#666;margin-bottom:10px;">⏱️ Elapsed: 00:00</div>
     <script>
+    var timerInterval;
+    
     function updateTimer() {
         var start = Date.now();
         var timerElement = document.getElementById('timer');
         
-        var interval = setInterval(function() {
+        timerInterval = setInterval(function() {
             var elapsed = Date.now() - start;
             var minutes = Math.floor(elapsed / 60000);
             var seconds = Math.floor((elapsed % 60000) / 1000);
@@ -39,21 +44,33 @@ def timer():
                 (minutes < 10 ? '0' : '') + minutes + ':' + 
                 (seconds < 10 ? '0' : '') + seconds;
         }, 1000);
-        
-        // Cleanup when component is removed
-        return function() {
-            clearInterval(interval);
+    }
+    
+    function stopTimer() {
+        if (timerInterval) {
+            clearInterval(timerInterval);
+            document.getElementById('timer').style.color = '#00cc00';
         }
     }
-    var cleanup = updateTimer();
+    
+    updateTimer();
     
     // Handle Streamlit's component cleanup
     window.addEventListener('beforeunload', function() {
-        cleanup();
+        if (timerInterval) clearInterval(timerInterval);
     });
     </script>
     """
 
+# Stop timer function
+def stop_timer():
+    html("""
+    <script>
+    if (typeof stopTimer === 'function') {
+        stopTimer();
+    }
+    </script>
+    """, height=0)
 
 # Page setup
 st.set_page_config(page_title="Your Image to Audio Story", page_icon="🦜")
@@ -98,17 +115,19 @@ def text2audio(story_text):
 uploaded_file = st.file_uploader("Select an Image After the Models are Loaded...")
  
 if uploaded_file is not None:
+    # Save the image data to session state
+    bytes_data = uploaded_file.getvalue()
+    st.session_state.image_data = bytes_data
+    
     # Initialize progress containers
     status_text = st.empty()
     progress_bar = st.progress(0)
     
-     # Start JavaScript timer
+    # Start JavaScript timer
     html(timer(), height=50)
     
-
     try:
         # Save uploaded file
-        bytes_data = uploaded_file.getvalue()
         with open(uploaded_file.name, "wb") as file:
             file.write(bytes_data)
 
@@ -142,22 +161,30 @@ if uploaded_file is not None:
 
             # Final status and stop timer
             status_text.success("**✅ Generation complete!**")
-            html("<script>document.getElementById('timer').style.color = '#00cc00';</script>")
+            stop_timer()
 
         # Show results
-        # st.subheader("Results")
         st.write("**Caption:**", st.session_state.processed_data['scenario'])
         st.write("**Story:**", st.session_state.processed_data['story'])
 
     except Exception as e:
-        html("<script>document.getElementById('timer').remove();</script>")
+        stop_timer()
         status_text.error(f"**❌ Error:** {str(e)}")
         progress_bar.empty()
         raise e
-
     
     finally:
-        pass  # Timer cleanup handled by JavaScript
+        pass
+
+elif st.session_state.image_data is not None:
+    # Display the previously uploaded image from session state
+    st.image(st.session_state.image_data, caption="Uploaded Image", use_container_width=True)
+    
+    # Show previous results if available
+    if st.session_state.processed_data.get('scenario'):
+        st.write("**Caption:**", st.session_state.processed_data['scenario'])
+    if st.session_state.processed_data.get('story'):
+        st.write("**Story:**", st.session_state.processed_data['story'])
 
 
 # Audio playback