Update app.py

app.py

@@ -6,12 +6,13 @@ from pathlib import Path
 from datetime import datetime
 import os
 from dotenv import load_dotenv
+import time
 
 # Load OpenAI API key
 load_dotenv()
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# Initialize session state
+# Initialize session state for tasks
 if 'tasks' not in st.session_state:
     st.session_state.tasks = {
         day: [] for day in ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
@@ -26,27 +27,36 @@ st.sidebar.header("🎤 Voice Input")
 current_day = datetime.today().strftime('%A')
 voice_note = mic_recorder(start_prompt="🎤 Add Task", stop_prompt="⏹️ Stop", key='recorder')
 
-# Process voice input
+# Process voice input with retry logic
 if voice_note and 'bytes' in voice_note:
-    try:
-        # Save the recorded audio to a temporary file
-        audio_file = open("temp_audio.wav", "wb")
-        audio_file.write(voice_note['bytes'])
-        audio_file.close()
-        
-        # Transcribe using OpenAI Whisper
-        with open("temp_audio.wav", "rb") as audio_file:
-            transcript = client.audio.transcriptions.create(
-                model="whisper-1",
-                file=audio_file
-            )
-        
-        # Add transcribed task to current day
-        if transcript.text:
-            st.session_state.tasks[current_day].append({'text': transcript.text, 'checked': False})
-            st.sidebar.success(f"Added: {transcript.text}")
-    except Exception as e:
-        st.sidebar.error(f"Error: {str(e)}")
+    max_retries = 3  # Number of retries
+    retry_delay = 2  # Delay between retries (in seconds)
+
+    for attempt in range(max_retries):
+        try:
+            # Save the recorded audio to a temporary file
+            audio_file = open("temp_audio.wav", "wb")
+            audio_file.write(voice_note['bytes'])
+            audio_file.close()
+            
+            # Transcribe using OpenAI Whisper
+            with open("temp_audio.wav", "rb") as audio_file:
+                transcript = client.audio.transcriptions.create(
+                    model="whisper-1",
+                    file=audio_file
+                )
+            
+            # Add transcribed task to current day
+            if transcript.text:
+                st.session_state.tasks[current_day].append({'text': transcript.text, 'checked': False})
+                st.sidebar.success(f"Added: {transcript.text}")
+            break  # Exit the retry loop if successful
+        except Exception as e:
+            if attempt < max_retries - 1:  # Don't wait on the last attempt
+                st.sidebar.warning(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds...")
+                time.sleep(retry_delay)
+            else:
+                st.sidebar.error(f"Error: {str(e)}")
 
 # Progress bar
 completed_tasks = sum(