Update app.py

app.py

@@ -35,11 +35,26 @@ def timer():
     // Create new timer element in parent document
     var timerDiv = window.parent.document.createElement('div');
     timerDiv.id = 'parent-timer';
-    timerDiv.style.fontSize = '16px';
-    timerDiv.style.color = '#666';
-    timerDiv.style.marginBottom = '10px';
+    timerDiv.style.cssText = `
+        font-size: 16px;
+        color: #666;
+        margin: 10px 0;
+        padding: 8px;
+        background: #f0f2f6;
+        border-radius: 4px;
+        position: sticky;
+        top: 0;
+        z-index: 999;
+    `;
     timerDiv.innerHTML = '⏱️ Elapsed: 00:00';
-    window.parent.document.body.appendChild(timerDiv);
+    
+    // Insert timer just below the header
+    var header = window.parent.document.querySelector('[data-testid="stHeader"]');
+    if (header) {
+        header.insertAdjacentElement('afterend', timerDiv);
+    } else {
+        window.parent.document.body.prepend(timerDiv);
+    }
     
     var start = Date.now();
     var timerInterval = setInterval(function() {
@@ -153,7 +168,7 @@ if uploaded_file is not None:
             """)
 
         # Show results
-        st.subheader("Results")
+        # st.subheader("Results")
         st.write("**Caption:**", st.session_state.processed_data['scenario'])
         st.write("**Story:**", st.session_state.processed_data['story'])