Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-import time
+import speech_recognition as sr
 from transformers import pipeline
 
-st.title("Enhanced Multilingual Translator Chatbot")
+st.title("Speech-to-Text and Translation Chatbot")
 
 # Choose the translation models from Hugging Face
 translation_models = {
@@ -12,10 +12,6 @@ translation_models = {
     "French to English": "Helsinki-NLP/opus-mt-fr-en",
     "English to Urdu": "Helsinki-NLP/opus-mt-en-ur",
     "Urdu to English": "Helsinki-NLP/opus-mt-ur-en",
-    "English to Spanish": "Helsinki-NLP/opus-mt-en-es",
-    "Spanish to English": "Helsinki-NLP/opus-mt-es-en",
-    "English to Chinese": "Helsinki-NLP/opus-mt-en-zh",
-    "Chinese to English": "Helsinki-NLP/opus-mt-zh-en",
     # Add more language pairs as needed
 }
 
@@ -24,33 +20,28 @@ selected_translation = st.selectbox("Select translation model", list(translation
 # Load the translation pipeline
 translator = pipeline(task="translation", model=translation_models[selected_translation])
 
-# User input for translation
-user_input = st.text_area("Enter text for translation:", "")
-
-# Display loading indicator
-if st.button("Translate"):
-    with st.spinner("Translating..."):
-        # Simulate translation delay for demonstration
-        time.sleep(2)
-        if user_input:
-            # Perform translation
-            translated_text = translator(user_input, max_length=500)[0]['translation_text']
-            st.success(f"Translated Text: {translated_text}")
-        else:
-            st.warning("Please enter text for translation.")
-
-# Clear button to reset input and result
-if st.button("Clear"):
-    user_input = ""
-    st.success("Input cleared.")
-    st.empty()  # Clear previous results if any
-
-st.markdown("---")
-
-st.subheader("About")
-st.write(
-    "This is an enhanced Multilingual Translator chatbot that uses the Hugging Face Transformers library."
-)
-st.write(
-    "Select a translation model from the dropdown, enter text, and click 'Translate' to see the translation."
-)
+# Function to perform speech-to-text and translation
+def speech_to_text_and_translate():
+    recognizer = sr.Recognizer()
+
+    st.info("Speak into your microphone...")
+
+    with sr.Microphone() as source:
+        audio_data = recognizer.listen(source, timeout=10)
+
+    try:
+        text = recognizer.recognize_google(audio_data)
+        st.success(f"Speech-to-Text Result: {text}")
+
+        # Perform translation
+        translated_text = translator(text, max_length=500)[0]['translation_text']
+        st.success(f"Translated Text: {translated_text}")
+
+    except sr.UnknownValueError:
+        st.warning("Speech recognition could not understand audio.")
+    except sr.RequestError as e:
+        st.error(f"Error with the speech recognition service: {e}")
+
+# Use speech-to-text and translation
+if st.button("Start Speech-to-Text and Translation"):
+    speech_to_text_and_translate()