Update app.py

app.py

@@ -51,25 +51,26 @@ def process_audio_input(audio, whisper_processor, whisper_model):
 
 # Generate response within a GPU-decorated function
 @spaces.GPU
-def generate_response(text_input, sarvam_pipe):
-    if sarvam_pipe is None:
-        return "Error: sarvam-2b model is not available. The assistant cannot generate responses at this time."
-    
-    try:
-        response = sarvam_pipe(text_input, max_new_tokens=100, temperature=0.7, repetition_penalty=1.2)[0]['generated_text']
-        return response
-    except Exception as e:
-        return f"Error generating response: {str(e)}"
-
 def text_to_speech(text, lang='hi'):
     try:
-        tts = gTTS(text=text, lang=lang)
+        # Use a better TTS engine for Indic languages
+        if lang in ['hi', 'bn', 'gu', 'kn', 'ml', 'mr', 'or', 'pa', 'ta', 'te']:
+            # You might want to use a different TTS library here
+            # For example, you could use the Google Cloud Text-to-Speech API
+            # or a specialized Indic language TTS library
+            
+            # This is a placeholder for a better Indic TTS solution
+            tts = gTTS(text=text, lang=lang, tld='co.in')  # Use Indian TLD
+        else:
+            tts = gTTS(text=text, lang=lang)
+        
         tts.save("response.mp3")
         return "response.mp3"
     except Exception as e:
         print(f"Error in text-to-speech: {str(e)}")
         return None
 
+# Replace the existing detect_language function with this improved version
 def detect_language(text):
     lang_codes = {
         'bn': 'Bengali', 'gu': 'Gujarati', 'hi': 'Hindi', 'kn': 'Kannada',
@@ -77,10 +78,15 @@ def detect_language(text):
         'ta': 'Tamil', 'te': 'Telugu', 'en': 'English'
     }
     
-    for code, lang in lang_codes.items():
-        if any(word in text for word in ['नमस्ते', 'હેલો', 'ನಮಸ್ಕಾರ', 'ഹലോ', 'नमस्कार', 'ਸਤ ਸ੍ਰੀ ਅਕਾਲ', 'வணக்கம்', 'నమస్కారం']):
-            return 'hi'  # Default to Hindi for simplicity
-    return 'en'  # Default to English if no Indic script is detected
+    try:
+        detected_lang = detect(text)
+        return detected_lang if detected_lang in lang_codes else 'en'
+    except:
+        # Fallback to simple script-based detection
+        for code, lang in lang_codes.items():
+            if any(ord(char) >= 0x0900 and ord(char) <= 0x097F for char in text):  # Devanagari script
+                return 'hi'
+        return 'en'  # Default to English if no Indic script is detected
 
 @spaces.GPU
 def indic_language_assistant(input_type, audio_input, text_input):