Update app.py

app.py

@@ -1,97 +1,36 @@
 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}"
+# API URL
+API_URL = "https://startrz-proagents.hf.space/api/v1/prediction/c4349321-8eba-43b6-a5ee-071fd56b8cd1"
 
-def query_prediction_api(audio_base64):
-    """Send request to prediction API"""
-    url = "https://startrz-proagents.hf.space/api/v1/prediction/9ad2e52d-052f-44e5-b3a0-7fd432d86506"
-    
-    payload = {
-        "uploads": [
-            {
-                "data": audio_base64,
-                "type": "audio",
-                "name": "audio.wav",
-                "mime": "audio/webm"
-            }
-        ], 
-        "overrideConfig": {
-            "sessionId": "test"
-        } 
-    }
-    
+# Function to query the API
+def query_api(question):
+    payload = {"question": question}
     try:
-        response = requests.post(
-            url,
-            headers={"Content-Type": "application/json"},
-            json=payload
-        )
-        response.raise_for_status()
-        return response.json()
+        response = requests.post(API_URL, json=payload)
+        response.raise_for_status()  # Raise an error for bad responses
+        return response.json()  # Return the JSON response
     except requests.exceptions.RequestException as e:
-        st.error(f"Error making prediction: {str(e)}")
-        return None
+        return {"error": str(e)}  # Return the error message
 
-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)
+# Streamlit app title
+st.title("API Query App")
 
-    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
-    """)
+# User input for the question
+user_question = st.text_input("Ask a question:")
+
+# Submit button
+if st.button("Submit"):
+    if user_question:
+        with st.spinner("Fetching response..."):
+            output = query_api(user_question)
+            if "error" in output:
+                st.error(f"Error: {output['error']}")
+            else:
+                # Display the response from the API
+                st.subheader("Response:")
+                st.write(output)  # Adjust this line based on the expected response structure
+    else:
+        st.warning("Please enter a question before submitting.")
 
-if __name__ == "__main__":
-    main()