Update app.py

app.py

@@ -15,13 +15,10 @@ from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from datetime import datetime
 from google.oauth2.service_account import Credentials
-from gtts import gTTS  # Text-to-speech
-import speech_recognition as sr  # Speech-to-text
+from gtts import gTTS
 import tempfile
 import base64
-from io import BytesIO# For temporary audio files
-
-
+from io import BytesIO
 
 
 
@@ -138,21 +135,6 @@ BATTERY_COLORS = {
 SPONSOR = "default"  # Set this to the sponsor name or leave as "default" for default branding
 TRAINER_SHEET_ID = "1GiA8pxZn04aUA-OKwcANvfJ_CpChooF2mUQxiJ_i2-s"
 
-def text_to_speech(text):
-    """Convert text to speech and return base64 encoded audio data"""
-    try:
-        # Create speech from text
-        tts = gTTS(text=text, lang='en')
-        
-        # Save to temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
-            tts.save(fp.name)
-            return fp.name
-            
-    except Exception as e:
-        print(f"Error in text-to-speech: {str(e)}")
-        return None
-
 def record_trainer_info(email, client_email, freelancer_email, freelancer_link, seal):
     """Record trainer information to Google Sheet"""
     try:
@@ -2889,43 +2871,70 @@ button:active {
 .progress-bar {
     display: none !important;
 }
-
-.voice-btn {
-    min-width: 40px !important;
-    height: 40px !important;
-    border-radius: 50% !important;
-    padding: 0 !important;
-    margin-left: 5px !important;
-    background: linear-gradient(135deg, #6e8efb, #a777e3) !important;
-    color: white !important;
+"""
+audio_player_css = """
+.audio-player-container {
+    display: flex;
+    align-items: center;
+    gap: 5px;
+    margin-left: 5px;
 }
-.voice-btn:hover {
-    transform: scale(1.05) !important;
+.audio-player-btn {
+    background: none;
+    border: none;
+    cursor: pointer;
+    font-size: 16px;
+    padding: 5px;
 }
-/* Audio player styling */
-.gradio-audio {
-    margin-top: 10px;
-    background: rgba(255, 255, 255, 0.7) !important;
-    border-radius: 8px !important;
-    padding: 10px !important;
+.audio-player-btn:hover {
+    color: #4CAF50;
 }
-
-/* Voice button styling */
-.voice-btn {
-    min-width: 40px !important;
-    height: 40px !important;
-    border-radius: 50% !important;
-    padding: 0 !important;
-    margin-left: 5px !important;
-    background: linear-gradient(135deg, #6e8efb, #a777e3) !important;
-    color: white !important;
+.download-btn {
+    background: none;
+    border: none;
+    cursor: pointer;
+    font-size: 16px;
+    padding: 5px;
+    color: #666;
 }
-.voice-btn:hover {
-    transform: scale(1.05) !important;
+.download-btn:hover {
+    color: #4CAF50;
 }
+"""
 
+# Add this JavaScript for audio control
+audio_player_js = """
+function createAudioPlayer(audioData) {
+    if (!audioData) return '';
+    
+    const audioId = 'audio-' + Math.random().toString(36).substr(2, 9);
+    const downloadUrl = 'data:audio/mp3;base64,' + audioData;
+    
+    return `
+    <div class="audio-player-container">
+        <audio id="${audioId}" src="${downloadUrl}"></audio>
+        <button class="audio-player-btn" onclick="document.getElementById('${audioId}').play()">▶️</button>
+        <button class="audio-player-btn" onclick="document.getElementById('${audioId}').pause()">⏸️</button>
+        <button class="audio-player-btn" onclick="document.getElementById('${audioId}').currentTime = 0; document.getElementById('${audioId}').pause()">⏹️</button>
+        <a class="download-btn" href="${downloadUrl}" download="response.mp3">💾</a>
+    </div>
+    `;
+}
 """
-    
+
+def text_to_speech(text):
+    """Convert text to speech and return base64 encoded audio"""
+    try:
+        tts = gTTS(text=text, lang='en', slow=False)
+        with tempfile.NamedTemporaryFile(delete=True, suffix='.mp3') as fp:
+            tts.save(fp.name)
+            with open(fp.name, 'rb') as audio_file:
+                audio_bytes = audio_file.read()
+        return base64.b64encode(audio_bytes).decode('utf-8')
+    except Exception as e:
+        print(f"Error in text-to-speech: {str(e)}")
+        return None
+
 def respond(message, history, max_tokens, temperature, top_p):
     email = OWNER_EMAIL
     
@@ -2933,12 +2942,10 @@ def respond(message, history, max_tokens, temperature, top_p):
     save_chat_history(email, message, "")
 
     try:
-        # Save to BATTERY_SHEET_ID (using our multi-sheet function)
+        # Save to BATTERY_SHEET_ID
         sheet = get_or_create_sheet(BATTERY_SHEET_ID, "ChatHistory")
-        
         now_str = datetime.now().strftime("%m/%d/%Y %H:%M:%S")
         sheet.append_row([now_str, email, message, ""])
-    
     except Exception as e:
         print(f"Error saving to battery sheet: {str(e)}")
     
@@ -2946,7 +2953,7 @@ def respond(message, history, max_tokens, temperature, top_p):
     temp_response = f"""<a href="https://flutterwave.com/pay/gossapp" target="_blank">
         <img src='https://i.imgur.com/PznT4qo.png' width='20'>
     </a>"""
-    yield [(message, temp_response)], None  # Add None for audio output
+    yield [(message, temp_response)], None  # None for initial audio
 
     try:
         # Build prompt
@@ -2955,13 +2962,12 @@ def respond(message, history, max_tokens, temperature, top_p):
         for user_msg, bot_msg in history:
             if user_msg:
                 messages.append({"role": "user", "content": user_msg})
-            if bot_msg and not str(bot_msg).startswith("<img"):  # Convert to string before checking
+            if bot_msg and not str(bot_msg).startswith("<img"):
                 messages.append({"role": "assistant", "content": str(bot_msg)})
         messages.append({"role": "user", "content": message})
 
-        # Get the current inference client
+        # Get response
         inference_client = get_inference_client()
-
         response = ""
         for chunk in inference_client.chat_completion(
             messages,
@@ -2974,36 +2980,33 @@ def respond(message, history, max_tokens, temperature, top_p):
             if token:
                 response += token
                 clean_response = response.split("(Note:")[0].strip()
-                yield [(message, clean_response)], None  # Add None for audio output
+                yield [(message, clean_response)], None  # No audio during streaming
 
         clean_response = response.split("(Note:")[0].strip()
-
-        # Generate audio for the response
-        audio_file = text_to_speech(clean_response)
         
-        # Save full bot response to chat history
+        # Generate audio
+        audio_data = text_to_speech(clean_response)
+        
+        # Save to chat history
         save_chat_history(email, message, clean_response)
 
-        # Update response in battery sheet
+        # Update battery sheet
         try:
             scope = ['https://www.googleapis.com/auth/spreadsheets',
                     'https://www.googleapis.com/auth/drive']
             creds = ServiceAccountCredentials.from_json_keyfile_name('credentials.json', scope)
-            gspread_client = gspread.authorize(creds)  # Renamed to avoid conflict
+            gspread_client = gspread.authorize(creds)
             
             spreadsheet = gspread_client.open_by_key(BATTERY_SHEET_ID)
             
-            # Search all sheets for our record to update
             for worksheet in spreadsheet.worksheets():
                 try:
                     records = worksheet.get_all_records()
-                    for i, record in enumerate(records, start=2):  # start=2 because header is row 1
+                    for i, record in enumerate(records, start=2):
                         if (str(record.get('Email', '')).lower() == email.lower() and 
                             str(record.get('User Message', '')).strip() == message.strip() and 
                             (not record.get('Bot Message') or str(record.get('Bot Message', '')).strip() == "")):
-                            
-                            # Update the bot response
-                            worksheet.update_cell(i, 4, clean_response)  # Column D is index 4
+                            worksheet.update_cell(i, 4, clean_response)
                             break
                 except Exception as e:
                     print(f"Error searching sheet {worksheet.title}: {str(e)}")
@@ -3012,7 +3015,7 @@ def respond(message, history, max_tokens, temperature, top_p):
         except Exception as e:
             print(f"Error updating battery sheet with response: {str(e)}")
 
-        yield [(message, clean_response)], audio_file if audio_file else None
+        yield [(message, clean_response)], audio_data if audio_data else None
 
     except Exception as e:
         if "402" in str(e):
@@ -3022,8 +3025,8 @@ def respond(message, history, max_tokens, temperature, top_p):
      Please:<br>
     1. <a href="https://flutterwave.com/pay/gossapp" target="_blank">Reload AI-time</a><br>
     2. Try again later,or<br>
-    3. Contact <a href="https://wa.link/x2wdpl" target="_blank">Support </a> if this persists.
-    Tired of Manual Chatbot reset delays? Consider a Subscription to have your chatbot last upto 20 times longer and fewer/ zero reset delays (Auto-reset)</div>
+    3. Contact <a href="https://wa.link/x2wdpl" target="_blank">Support </a> if this persists.<br><br>
+    Tired of Manual Chatbot reset delays? <br>Consider a Subscription to have your chatbot last upto 20 times longer and fewer/ zero reset delays (Auto-reset)</div>
             """)
         else:
             custom_error = gr.HTML(f"""
@@ -3316,52 +3319,34 @@ with gr.Blocks(
                
 
         
-            with gr.Column(scale=3, elem_classes="chat-container"):
-                # Create the HTML component separately
-                greeting_html = gr.HTML(get_chatbot_greeting())
-                
-                chatbot = gr.Chatbot(
-                    elem_id="chatbot",
+        with gr.Column(scale=3, elem_classes="chat-container"):
+            # Create the HTML component separately
+            greeting_html = gr.HTML(get_chatbot_greeting())
+            
+            chatbot = gr.Chatbot(
+                elem_id="chatbot",
+                show_label=False,
+                avatar_images=(None, user_data['avatar_url']),
+                show_copy_button=False,
+                show_share_button=False,
+                layout="panel",
+                value=[(None, greeting_html)],
+                height=600,
+                # Add these to ensure proper centering in the container
+                elem_classes=["centered-chatbot"]
+            )
+            
+            # Replace the existing button row and clear button code with this:
+            with gr.Row():
+                msg = gr.Textbox(
                     show_label=False,
-                    avatar_images=(None, user_data['avatar_url']),
-                    show_copy_button=False,
-                    show_share_button=False,
-                    layout="panel",
-                    value=[(None, greeting_html)],
-                    height=600,
+                    placeholder="Let's GossApp...",
+                    container=False,
+                    autofocus=True,
+                    scale=6
                 )
-                
-                with gr.Row():
-                    with gr.Column(scale=6):
-                        msg = gr.Textbox(
-                            show_label=False,
-                            placeholder="Let's GossApp! Type or say something...",
-                            container=False,
-                            autofocus=True
-                        )
-                    
-                    with gr.Column(scale=1, min_width=50):
-                        voice_btn = gr.Button("🎤", elem_classes="voice-btn")
-                    
-                    submit = gr.Button("", elem_classes="send-btn")
-                
-                # Audio output for TTS responses
-                audio_output = gr.Audio(visible=True, label="Chatbot Response", autoplay=True)
-                
-                clear = gr.ClearButton([msg, chatbot], variant="secondary")
-            
-            with gr.Accordion("🔊 Alternative: Upload Audio", open=False):
-                    audio_upload = gr.Audio(
-                        sources=["upload"],  # Allow file upload
-                        type="filepath",
-                        label="Or upload an audio file"
-                    )
-                    
-                    # Process uploaded audio
-                    audio_upload.change(
-                        fn=lambda audio: listen_microphone(audio),  # Reuse speech-to-text
-                        outputs=msg
-                    )
+                submit = gr.Button("", elem_classes="send-btn")
+                audio_output = gr.HTML(visible=False) 
             
             # Add Business, Family, Friends chatbots right below the button row
             business_html_display = gr.HTML(f"""
@@ -5347,17 +5332,30 @@ with gr.Blocks(
         outputs=[save_status, bio_display, chatbot]
     )
     
-    submit.click(
+    # Update your submit/msg handlers
+    msg.submit(
         respond,
-        inputs=[msg, chatbot, max_tokens, temperature],
-        outputs=[chatbot, audio_output],
-    ).then(clear_input, outputs=[msg])
+        inputs=[msg, chatbot, max_tokens, temperature, top_p],
+        outputs=[chatbot, audio_output]
+    ).then(
+        lambda: gr.HTML(visible=True),
+        outputs=[audio_output]
+    ).then(
+        clear_input,
+        outputs=[msg]
+    )
     
-    msg.submit(
+    submit.click(
         respond,
-        inputs=[msg, chatbot, max_tokens, temperature],
-        outputs=[chatbot, audio_output],
-    ).then(clear_input, outputs=[msg])
+        inputs=[msg, chatbot, max_tokens, temperature, top_p],
+        outputs=[chatbot, audio_output]
+    ).then(
+        lambda: gr.HTML(visible=True),
+        outputs=[audio_output]
+    ).then(
+        clear_input,
+        outputs=[msg]
+    )
 
     # Add these at the end of your existing event handlers
     submit_complaint_btn.click(
@@ -5372,7 +5370,24 @@ with gr.Blocks(
         outputs=[status_display]
     )
 
-    # Certificate generation
+# Add the audio player update function
+def update_audio_player(audio_data):
+    if audio_data:
+        return f"""
+        <script>
+            {audio_player_js}
+            document.currentScript.parentNode.innerHTML = createAudioPlayer('{audio_data}');
+        </script>
+        """
+    return ""
+
+    audio_output.change(
+        update_audio_player,
+        inputs=[audio_output],
+        outputs=[audio_output]
+    )
+
+# Certificate generation
     def generate_certificate(name, email):
         """Generate certificate HTML with centered name on background image"""
 
@@ -5415,28 +5430,6 @@ with gr.Blocks(
         outputs=[cert_display]
     )
 
-def listen_microphone():
-    r = sr.Recognizer()
-    try:
-        with sr.Microphone() as source:
-            print("Listening...")
-            audio = r.listen(source, timeout=5)
-            user_text = r.recognize_google(audio)
-            return user_text
-    except Exception as e:
-        raise gr.Error(f"Error: {str(e)}. Try typing instead.")
-
-    voice_btn.click(
-        listen_microphone,
-        outputs=[msg]
-    )
-
-def speak_response(text):
-    with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as fp:
-        tts = gTTS(text=text, lang="en")
-        tts.save(fp.name)
-        return fp.name
-    
 def auto_refresh():
     while True:
         # Check for inactive accounts once per day