Update app.py

app.py

@@ -10,6 +10,7 @@ import uuid
 import pandas as pd
 import plotly.express as px
 import streamlit as st
+import pyaudio
 
 # Initialize components
 objection_handler = ObjectionHandler('objections.csv')
@@ -17,30 +18,35 @@ product_recommender = ProductRecommender('recommendations.csv')
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
 def list_audio_devices():
-    """List available audio input devices."""
-    recognizer = sr.Recognizer()
-    mic_list = sr.Microphone.list_microphone_names()
-    return mic_list
+    """List available audio input devices using pyaudio."""
+    p = pyaudio.PyAudio()
+    devices = []
+    for i in range(p.get_device_count()):
+        device_info = p.get_device_info_by_index(i)
+        if device_info["maxInputChannels"] > 0:  # Check if it's an input device
+            devices.append(device_info)
+    return devices
 
 def real_time_analysis():
     st.info("Listening... Say 'stop' to end the process.")
 
-    recognizer = sr.Recognizer()
-    sentiment_scores = []
-    transcribed_chunks = []
-    total_text = ""
-
     try:
         # List available audio devices
-        mic_list = list_audio_devices()
-        st.write(f"Available audio devices: {mic_list}")
+        devices = list_audio_devices()
+        st.write("Available audio devices:")
+        for device in devices:
+            st.write(f"Device {device['index']}: {device['name']} (Input Channels: {device['maxInputChannels']})")
 
-        if not mic_list:
-            st.error("No audio devices found. Please check the virtual microphone setup.")
+        if not devices:
+            st.error("No audio input devices found. Please check the virtual microphone setup.")
             return
 
-        # Use the virtual microphone (device index 0)
-        mic = sr.Microphone(device_index=0)  # Use the virtual microphone device index
+        # Use the first available input device
+        device_index = devices[0]["index"]
+        st.write(f"Using device index {device_index}: {devices[0]['name']}")
+
+        recognizer = sr.Recognizer()
+        mic = sr.Microphone(device_index=device_index)
 
         while True:
             with mic as source: