Update app.py

app.py

@@ -117,7 +117,7 @@ def handle_objection(text):
         return "\n".join(responses) if responses else "No objection response found."
     return "No objection response found."
 
-def transcribe_audio(audio_bytes):
+def transcribe_audio(audio_bytes, sample_rate=16000):
     """Transcribe audio using the transcribe_with_chunks function from sentiment_analysis.py."""
     try:
         # Save audio bytes to a temporary WAV file
@@ -125,7 +125,7 @@ def transcribe_audio(audio_bytes):
             with wave.open(wav_buffer, 'wb') as wf:
                 wf.setnchannels(1)  # Mono audio
                 wf.setsampwidth(2)  # 2 bytes for int16
-                wf.setframerate(16000)  # Sample rate
+                wf.setframerate(sample_rate)  # Sample rate
                 wf.writeframes(audio_bytes)
 
             # Use the transcribe_with_chunks function from sentiment_analysis.py
@@ -140,37 +140,38 @@ def real_time_analysis():
     st.info("Listening... Say 'stop' to end the process.")
 
     def audio_frame_callback(audio_frame):
-        # Convert audio frame to bytes
-        audio_bytes = audio_frame.to_ndarray().tobytes()
+    # Convert audio frame to bytes
+    audio_data = audio_frame.to_ndarray()
+    audio_bytes = (audio_data * 32767).astype(np.int16).tobytes()  # Convert to int16 format
 
-        # Transcribe the audio
-        text = transcribe_audio(audio_bytes)
-        if text:
-            st.write(f"*Recognized Text:* {text}")
+    # Transcribe the audio
+    text = transcribe_audio(audio_bytes)
+    if text:
+        st.write(f"*Recognized Text:* {text}")
 
-            # Analyze sentiment
-            sentiment, score = analyze_sentiment(text)
-            st.write(f"*Sentiment:* {sentiment} (Score: {score})")
+        # Analyze sentiment
+        sentiment, score = analyze_sentiment(text)
+        st.write(f"*Sentiment:* {sentiment} (Score: {score})")
 
-            # Handle objection
-            objection_response = handle_objection(text)
-            st.write(f"*Objection Response:* {objection_response}")
+        # Handle objection
+        objection_response = handle_objection(text)
+        st.write(f"*Objection Response:* {objection_response}")
 
-            # Get product recommendation
-            recommendations = []
-            if is_valid_input(text) and is_relevant_sentiment(score):
-                query_embedding = model.encode([text])
-                distances, indices = product_recommender.index.search(query_embedding, 1)
+        # Get product recommendation
+        recommendations = []
+        if is_valid_input(text) and is_relevant_sentiment(score):
+            query_embedding = model.encode([text])
+            distances, indices = product_recommender.index.search(query_embedding, 1)
 
-                if distances[0][0] < 1.5:  # Similarity threshold
-                    recommendations = product_recommender.get_recommendations(text)
+            if distances[0][0] < 1.5:  # Similarity threshold
+                recommendations = product_recommender.get_recommendations(text)
 
-            if recommendations:
-                st.write("*Product Recommendations:*")
-                for rec in recommendations:
-                    st.write(rec)
+        if recommendations:
+            st.write("*Product Recommendations:*")
+            for rec in recommendations:
+                st.write(rec)
 
-        return audio_frame
+    return audio_frame
 
     # Start WebRTC audio stream
     webrtc_ctx = webrtc_streamer(
@@ -271,4 +272,5 @@ def run_app():
             st.error(f"Error loading dashboard: {e}")
 
 if __name__ == "__main__":
-    run_app()
+    run_app()
+