Update app.py

app.py

@@ -9,7 +9,6 @@ from io import BytesIO
 from md2pdf.core import md2pdf
 from dotenv import load_dotenv
 from datetime import datetime
-import backoff
 import requests
 from requests.exceptions import RequestException
 from typing import Optional, Dict, Any
@@ -20,7 +19,7 @@ from st_audiorec import st_audiorec  # Import the audio recorder component
 MAX_FILE_SIZE = 41943040  # 40MB
 FILE_TOO_LARGE_MESSAGE = "File too large. Maximum size is 40MB."
 
-# Load environment variables
+# Load environment variables - will also load from Hugging Face secrets
 load_dotenv()
 
 # Initialize session states
@@ -48,7 +47,7 @@ st.set_page_config(
 LLM_MODEL = "deepseek-r1-distill-llama-70b"
 TRANSCRIPTION_MODEL = "distil-whisper-large-v3-en"
 
-# Configure exponential backoff for API retries
+# Configure retry parameters without backoff library
 MAX_RETRIES = 5
 INITIAL_WAIT = 0.5
 MAX_WAIT = 30
@@ -192,16 +191,39 @@ class NoteSection:
                 markdown_content += self.get_markdown_content(content, level + 1)
         return markdown_content
 
-# Add backoff decorator for retrying transcription
-@backoff.on_exception(
-    backoff.expo,
-    (RequestException, GroqAPIError),
-    max_tries=MAX_RETRIES,
-    factor=INITIAL_WAIT,
-    max_value=MAX_WAIT,
-    jitter=backoff.full_jitter,
-    on_backoff=lambda details: st.info(f"Retrying transcription... (Attempt {details['tries']}/{MAX_RETRIES})")
-)
+# Manual implementation of retry logic (replacing backoff library)
+def retry_with_exponential_backoff(max_tries=MAX_RETRIES, initial_wait=INITIAL_WAIT, max_wait=MAX_WAIT):
+    """
+    Custom retry decorator with exponential backoff
+    """
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            wait = initial_wait
+            tries = 0
+            
+            while tries < max_tries:
+                try:
+                    return func(*args, **kwargs)
+                except (RequestException, GroqAPIError) as e:
+                    tries += 1
+                    if tries == max_tries:
+                        raise
+                    
+                    # Calculate next wait time with exponential backoff
+                    wait = min(wait * 2, max_wait)
+                    
+                    # Add some jitter (±20%)
+                    jitter = wait * 0.4 * (np.random.random() - 0.5)
+                    wait_with_jitter = wait + jitter
+                    
+                    # Inform user of retry attempt
+                    st.info(f"Retrying transcription... (Attempt {tries}/{max_tries})")
+                    time.sleep(wait_with_jitter)
+                    
+        return wrapper
+    return decorator
+
+@retry_with_exponential_backoff()
 def transcribe_audio_with_groq(audio_data) -> str:
     """
     Transcribe audio file using Groq's transcription API with retry capability
@@ -462,56 +484,21 @@ def main():
     st.title("🧙‍♂️ ScribeWizard")
     st.markdown("Transform speech into highly structured notes with AI magic")
     
-    # Sidebar for configuration
+    # Initialize API key from environment (Hugging Face secrets)
+    if st.session_state.api_key and not st.session_state.groq_client:
+        st.session_state.groq_client = initialize_groq_client(st.session_state.api_key)
+    
+    # Display model info in the sidebar
     with st.sidebar:
-        st.header("Configuration")
-        
-        # Improved API key handling
-        api_key_input = st.text_input(
-            "Groq API Key",
-            value=st.session_state.api_key,
-            type="password",
-            help="Enter your Groq API key. Get one at https://console.groq.com/keys"
-        )
-        
-        # Only update if changed
-        if api_key_input != st.session_state.api_key:
-            st.session_state.api_key = api_key_input
-            # Reinitialize client with new key
-            st.session_state.groq_client = initialize_groq_client(api_key_input)
-            
-            if st.session_state.groq_client:
-                st.success("✅ API key validated!")
-            
-        # Initialize client if not done already
-        if st.session_state.api_key and not st.session_state.groq_client:
-            st.session_state.groq_client = initialize_groq_client(st.session_state.api_key)
-        
-        # API key instructions
-        with st.expander("How to get a Groq API key"):
-            st.markdown("""
-            1. Visit [Groq Cloud Console](https://console.groq.com/)
-            2. Sign up or log in to your account
-            3. Navigate to API Keys in the sidebar
-            4. Click "Create API Key" and give it a name
-            5. Copy the generated key (you won't see it again!)
-            
-            **Security tip**: For production use, store your API key as an environment variable:
-            ```bash
-            export GROQ_API_KEY="your-api-key-here"
-            ```
-            """)
-        
-        st.markdown("---")
         st.info("Using DeepSeek-R1-Distill-Llama-70B model for note generation and Distil Whisper for transcription")
     
     # Check if API key is valid before proceeding
     if not st.session_state.api_key:
-        st.warning("⚠️ Please enter your Groq API key in the sidebar to use ScribeWizard")
+        st.error("❌ No API key found. Please set the GROQ_API_KEY secret in your Hugging Face Space settings.")
         st.stop()
     
     if not st.session_state.groq_client:
-        st.error("❌ Failed to initialize Groq client. Please check your API key.")
+        st.error("❌ Failed to initialize Groq client. Please check your API key secret in Hugging Face Space settings.")
         st.stop()
     
     # Input methods tabs
@@ -618,7 +605,7 @@ def main():
                             if "Audio file too large" in str(e) or "exceeds" in str(e):
                                 st.info("💡 Try trimming your audio file or uploading a shorter segment.")
                             elif "API key" in str(e) or "Authentication" in str(e):
-                                st.info("💡 Check that your Groq API key is correct and has access to the transcription API.")
+                                st.info("💡 Check that your Groq API key is correct in your Hugging Face Space settings.")
                             elif "Rate limit" in str(e):
                                 st.info("💡 You've hit Groq's rate limits. Please wait a few minutes before trying again.")