Update app.py

app.py

@@ -12,71 +12,65 @@ import tempfile
 # Load Bhagavad Gita dataset
 df = pd.read_csv("bhagavad_gita.csv")
 
-# Load Sentence Transformer model and precompute embeddings
+# Load sentence transformer and precompute embeddings
 model = SentenceTransformer("all-MiniLM-L6-v2")
 verse_embeddings = model.encode(df['meaning_in_english'].tolist(), convert_to_tensor=True)
 
-# Temp audio files directory
+# Temp folder for audio files
 TEMP_DIR = tempfile.mkdtemp()
 
-# Background music path
+# Background music file path
 bg_music_path = "krishna_bg_music.mp3"
 
+def shorten_explanation(text, max_sentences=2):
+    sentences = text.split('. ')
+    shortened = '. '.join(sentences[:max_sentences]).strip()
+    if not shortened.endswith('.'):
+        shortened += '.'
+    return shortened
 
-# -------------------- NEW & FIXED FUNCTIONS -------------------- #
-
-def clean_text(text):
-    """Remove non-ASCII characters & extra spaces (fixes TTS crash)."""
+def clean_english(text):
     text = re.sub(r'[^\x00-\x7F]+', ' ', text)
     return ' '.join(text.split())
 
-
-async def generate_tts_async(text, voice="en-IN-PrabhatNeural"):
-    """Generate TTS audio using Microsoft Edge Neural Voice."""
-    filename = os.path.join(TEMP_DIR, f"krishna_voice_{uuid.uuid4()}.mp3")
-
+async def tts_generate_async(text, voice="en-IN-PrabhatNeural"):
+    """Generate audio asynchronously using edge-tts with Krishna male voice."""
+    filename = os.path.join(TEMP_DIR, f"voice_{uuid.uuid4()}.mp3")
     try:
-        communicate = edge_tts.Communicate(text, voice)
+        communicate = edge_tts.Communicate(text, voice=voice)
         await communicate.save(filename)
-        return filename
+        if os.path.exists(filename) and os.path.getsize(filename) > 1000:
+            return filename
     except Exception as e:
-        print("TTS Error:", e)
-        return None
-
+        print(f"TTS Error: {e}")
+    return None
 
 def generate_tts(text):
-    """Wrapper to run async TTS inside sync gradio fn."""
-    text = clean_text(text)
-
+    """Synchronous wrapper for async TTS generation."""
     if not text.strip():
         return None
-
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    return loop.run_until_complete(generate_tts_async(text))
-
-
-# --------------------------------------------------------------- #
-
-def shorten_explanation(text, max_sentences=2):
-    sentences = text.split('. ')
-    shortened = '. '.join(sentences[:max_sentences]).strip()
-    if not shortened.endswith('.'):
-        shortened += '.'
-    return shortened
-
+    try:
+        return asyncio.run(tts_generate_async(clean_english(text)))
+    except Exception as e:
+        print(f"TTS Generation failed: {e}")
+        return None
 
 def get_unique_bgm():
+    """Return unique copy of background music (optional)."""
     if not os.path.exists(bg_music_path):
         return None
     unique_bgm_path = os.path.join(TEMP_DIR, f"bgm_{uuid.uuid4()}.mp3")
-    shutil.copy(bg_music_path, unique_bgm_path)
-    return unique_bgm_path
-
+    try:
+        shutil.copy(bg_music_path, unique_bgm_path)
+        return unique_bgm_path
+    except Exception as e:
+        print(f"BGM Error: {e}")
+        return None
 
 def versewise_bot(question, play_music):
+    """Main chatbot logic: find the most similar verse and return the answer."""
     if not question.strip():
-        return "Please ask something meaningful.", None, None
+        return "Please ask a valid question.", None, None
 
     query_embedding = model.encode(question, convert_to_tensor=True)
     similarity_scores = util.pytorch_cos_sim(query_embedding, verse_embeddings)[0]
@@ -88,65 +82,71 @@ def versewise_bot(question, play_music):
     explanation = shorten_explanation(verse['meaning_in_english'])
     verse_number = verse['verse_number']
 
-    formatted_reply = f"""
-📿 **Bhagavad Gita {verse_number}**
+    reply = f"""📖 Bhagavad Gita {verse_number}
 
-🕉 *"{translation}"*
+🕉 "[translate:{sanskrit[:60]}...]"
 
-✨ {explanation}
+"{translation}"
 
-🌼 Remember: Krishna is walking with you.
-"""
-
-    # TEXT SENT TO TTS BASED ON YOUR NEW REQUIREMENT (NATURAL VOICE)
-    voice_text = f"{translation}. {explanation}"
+🕊 {explanation}
 
-    audio_voice = generate_tts(voice_text)
-    audio_bgm = get_unique_bgm() if play_music else None
+🌼 Stay strong — Krishna walks with you."""
 
-    return formatted_reply, audio_voice, audio_bgm
+    voice_text = f"{translation}. {explanation}"
+    audio_path = generate_tts(voice_text)
+    music_path = get_unique_bgm() if play_music else None
 
+    return reply, audio_path, music_path
 
 def get_quote_of_the_day():
     verse = df.sample(1).iloc[0]
-    return f"""
-<b>Quote of the Day (Gita {verse['verse_number']}):</b><br>
-<i>{verse['translation_in_english']}</i>
-"""
-
+    sanskrit = verse['verse_in_sanskrit']
+    translation = verse['translation_in_english']
+    verse_number = verse['verse_number']
+    return f"""<div style="font-size:1.1em;padding:10px 0;">
+    <b>Quote of the Day (Gita {verse_number}):</b><br>
+    <i>[translate:{sanskrit[:60]}...]</i><br>
+    <span style="color:#2d2d2d;">"{translation}"</span></div>"""
 
-# Custom Styling
+# Custom UI theme styling
 custom_css = """
 body, .gradio-container, .gradio-interface, html {
   background-image: url('https://static.vecteezy.com/system/resources/previews/022/592/272/large_2x/image-of-divine-beautiful-closed-eyes-blue-colored-krishna-generative-ai-free-photo.jpeg') !important;
-  background-size: cover;
+  background-size: cover !important;
+  background-repeat: no-repeat !important;
+  background-position: center center !important;
+  background-attachment: fixed !important;
 }
-.gradio-container {
-  background-color: rgba(255,255,255,0.90) !important;
+.gradio-container, .gradio-interface {
+  background-color: rgba(255,255,255,0.92) !important;
   border-radius: 18px;
   padding: 25px;
+  max-width: 760px;
+  box-shadow: 0 4px 15px rgba(0, 0, 0, 0.2);
+  margin: auto;
 }
 """
 
-# Gradio UI
+# Gradio interface without voice toggle; Krishna's male voice fixed
 interface = gr.Interface(
     fn=versewise_bot,
     inputs=[
-        gr.Textbox(label="Ask Krishna", placeholder="Why am I struggling?", lines=2),
+        gr.Textbox(label="Ask Krishna", placeholder="Why am I struggling in life?", lines=2),
         gr.Checkbox(label="Play Background Music", value=True)
     ],
     outputs=[
-        gr.Textbox(label="🧘 Krishna's Answer"),
+        gr.Textbox(label="🧘‍♂ Krishna's Answer"),
         gr.Audio(label="🔊 Krishna’s Voice", type="filepath"),
-        gr.Audio(label="🎶 Music", autoplay=True, type="filepath")
+        gr.Audio(label="🎶 Background Music", autoplay=True, type="filepath")
     ],
-    title="🕉 VerseWise - Divine Guidance from Bhagavad Gita",
-    description="Ask any question and receive a verse from the Bhagavad Gita.",
+    title="🕉 VerseWise - Divine Wisdom from the Gita",
+    description="Ask any question, and receive a verse from the Bhagavad Gita with divine guidance.",
     article=get_quote_of_the_day(),
     flagging_mode="never",
+    theme="soft",
     css=custom_css
 )
 
 if __name__ == "__main__":
-    print(f"Temp folder created: {TEMP_DIR}")
+    print(f"Temp directory for audio files: {TEMP_DIR}")
     interface.launch()