Update app.py

app.py

@@ -1,13 +1,22 @@
 import streamlit as st
-import os
-import time
-import re
-from openai import OpenAI
 from streamlit_webrtc import webrtc_streamer, AudioProcessorBase
 import av
 import numpy as np
 import tempfile
 import soundfile as sf
+import os
+import time
+import re
+from openai import OpenAI
+
+# ------------------ Audio Processor ------------------
+class AudioRecorder(AudioProcessorBase):
+    def __init__(self):
+        self.recorded_frames = []
+
+    def recv(self, frame: av.AudioFrame) -> av.AudioFrame:
+        self.recorded_frames.append(frame)
+        return frame
 
 # ------------------ App Configuration ------------------
 st.set_page_config(page_title="Document AI Assistant", layout="wide")
@@ -60,14 +69,6 @@ with col1:
         st.session_state.image_updated = False
 
 # ------------------ Voice Input Processing ------------------
-class AudioRecorder(AudioProcessorBase):
-    def __init__(self):
-        self.recorded_frames = []
-
-    def recv(self, frame: av.AudioFrame) -> av.AudioFrame:
-        self.recorded_frames.append(frame)
-        return frame
-
 with col2:
     st.markdown("### 🎙️ Voice Input (Optional)")
     webrtc_ctx = webrtc_streamer(
@@ -163,4 +164,4 @@ with col2:
 
     for message in reversed(st.session_state.messages):
         role, content = message["role"], message["content"]
-        st.chat_message(role).write(content)
+        st.chat_message(role).write(content)