Update requirements.txt

requirements.txt

@@ -1,3 +1,94 @@
-streamlit
-requests
-tiktoken
+import streamlit as st
+import requests
+import base64
+import json
+from io import BytesIO
+import numpy as np
+from audio_recorder_streamlit import audio_recorder
+
+def convert_audio_to_base64(audio_bytes):
+    """Convert audio bytes to base64 string"""
+    if audio_bytes is None:
+        return None
+    base64_str = base64.b64encode(audio_bytes).decode('utf-8')
+    return f"data:audio/webm;codecs=opus;base64,{base64_str}"
+
+def query_prediction_api(audio_base64):
+    """Send request to prediction API"""
+    url = "http://localhost:3000/api/v1/prediction/<chatlfowid>"
+    
+    payload = {
+        "uploads": [
+            {
+                "data": audio_base64,
+                "type": "audio",
+                "name": "audio.wav",
+                "mime": "audio/webm"
+            }
+        ]
+    }
+    
+    try:
+        response = requests.post(
+            url,
+            headers={"Content-Type": "application/json"},
+            json=payload
+        )
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        st.error(f"Error making prediction: {str(e)}")
+        return None
+
+def main():
+    st.title("Audio Prediction App")
+    st.write("Record or upload audio to get predictions")
+    
+    # Add tabs for different input methods
+    tab1, tab2 = st.tabs(["Record Audio", "Upload Audio"])
+    
+    with tab1:
+        st.write("Click the button below to record audio")
+        audio_bytes = audio_recorder()
+        
+        if audio_bytes:
+            st.audio(audio_bytes, format="audio/webm")
+            
+            if st.button("Get Prediction for Recorded Audio"):
+                with st.spinner("Getting prediction..."):
+                    audio_base64 = convert_audio_to_base64(audio_bytes)
+                    if audio_base64:
+                        prediction = query_prediction_api(audio_base64)
+                        if prediction:
+                            st.json(prediction)
+    
+    with tab2:
+        uploaded_file = st.file_uploader("Choose an audio file", type=['wav', 'mp3', 'webm'])
+        
+        if uploaded_file:
+            audio_bytes = uploaded_file.read()
+            st.audio(audio_bytes)
+            
+            if st.button("Get Prediction for Uploaded Audio"):
+                with st.spinner("Getting prediction..."):
+                    audio_base64 = convert_audio_to_base64(audio_bytes)
+                    if audio_base64:
+                        prediction = query_prediction_api(audio_base64)
+                        if prediction:
+                            st.json(prediction)
+
+    st.sidebar.markdown("""
+    ### About
+    This app allows you to:
+    - Record audio directly in the browser
+    - Upload audio files
+    - Get predictions from the API
+    
+    Supported formats:
+    - WAV
+    - MP3
+    - WebM
+    """)
+
+if __name__ == "__main__":
+    main()