Update app.py

app.py

@@ -1,43 +1,41 @@
 import streamlit as st
 import google.generativeai as genai
-import os
-from groq import Groq, GroqError
 import whisper
 from gtts import gTTS
 import tempfile
+import os
 import logging
 import numpy as np
 from pydub import AudioSegment
+from groq import Groq, GroqError
+import gradio as gr
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Securely configure Google API Key for generative AI (set this in the deployment environment)
+# Securely configure API Key (set this in the deployment environment)
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=GOOGLE_API_KEY)
 
-# Set up Groq API key for audio processing
-groq_api_key = os.getenv('groq_api_key')
-
-if not groq_api_key:
+# Set up Groq API key
+GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+if not GROQ_API_KEY:
     raise ValueError("GROQ_API_KEY is not set.")
 try:
-    groq_client = Groq(api_key=groq_api_key)
-    logger.info("Groq API key is set and client is initialized.")
+    groq_client = Groq(api_key=GROQ_API_KEY)
 except GroqError as e:
-    logger.error(f"Failed to initialize Groq client: {e}")
-    raise
+    raise ValueError(f"Failed to initialize Groq client: {e}")
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
+# Initialize Whisper Model
 try:
-    # Load Whisper model for audio transcription
     whisper_model = whisper.load_model("base")
     logger.info("Whisper model loaded successfully.")
 except Exception as e:
     logger.error(f"Failed to load Whisper model: {e}")
     raise
 
-# Initialize the Generative Model for heart health chatbot
+# Initialize the Generative Model
 model = genai.GenerativeModel(
     'gemini-1.5-flash',
     system_instruction=(
@@ -48,12 +46,7 @@ model = genai.GenerativeModel(
     )
 )
 
-# Function to get response from the Generative AI chatbot
-def get_chatbot_response(user_input):
-    response = model.generate_content(user_input)
-    return response.text.strip()
-
-# Function to process audio using Whisper, Groq, and gTTS
+# Function to process audio
 def process_audio(audio_file):
     try:
         # Transcribe audio using Whisper
@@ -92,7 +85,7 @@ def process_audio(audio_file):
 
 # Streamlit page configuration
 st.set_page_config(
-    page_title="Heart Health & Audio Chatbot",
+    page_title="Heart Health Chatbot",
     page_icon="👨‍⚕️",
     layout="centered",
     initial_sidebar_state="collapsed",
@@ -130,12 +123,12 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Custom header
+# Load and display a custom header
 def load_header():
     st.markdown("""
         <div style="padding:10px;text-align:center;color:white;">
-            <h1>Heart Health & Audio Chatbot 🫀</h1>
-            <p>Ask me anything about heart diseases or process audio queries!</p>
+            <h1>Heart Health Chatbot 🫀</h1>
+            <p>Ask me anything about heart diseases!</p>
         </div>
     """, unsafe_allow_html=True)
 
@@ -143,9 +136,8 @@ def load_header():
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# Avatar URLs
-user_avatar_url = "https://img.freepik.com/free-photo/sad-cartoon-anatomical-heart_23-2149767987.jpg"
-bot_avatar_url = "https://img.freepik.com/premium-photo/3d-render-man-doctor-avatar-round-sticker-with-cartoon-character-face-user-id-thumbnail-modern-69_1181551-3160.jpg"
+user_avatar_url = "https://img.freepik.com/free-photo/sad-cartoon-anatomical-heart_23-2149767987.jpg?t=st=1725263300~exp=1725266900~hmac=3763e175a896a554720d54c6d774dc645dd73078c952913accec719977d50b48&w=740"
+bot_avatar_url = "https://img.freepik.com/premium-photo/3d-render-man-doctor-avatar-round-sticker-with-cartoon-character-face-user-id-thumbnail-modern-69_1181551-3160.jpg?w=740"
 
 # Function to display chat history
 def display_chat_history():
@@ -172,11 +164,12 @@ def display_chat_history():
 # Main application layout
 def main():
     load_header()
+    st.write("")  # Add spacing
 
     with st.container():
         display_chat_history()
 
-        # User input for text-based chat
+        # User input area for text
         with st.form(key="user_input_form", clear_on_submit=True):
             user_input = st.text_input(
                 label="Type your message...",
@@ -188,23 +181,25 @@ def main():
             if submit_button and user_input.strip():
                 with st.spinner("Thinking..."):
                     bot_response = get_chatbot_response(user_input)
-                
+
                 # Update chat history
                 st.session_state.history.append({"role": "user", "content": user_input})
                 st.session_state.history.append({"role": "bot", "content": bot_response})
-                
+
                 # Refresh chat display
                 display_chat_history()
 
-        # File uploader for audio processing
-        st.subheader("Upload audio for processing")
-        audio_file = st.file_uploader("Choose an audio file", type=["mp3", "wav"])
+        # Audio upload section
+        st.markdown("### Upload your audio for consultation")
+        uploaded_audio = st.file_uploader("Upload an audio file", type=["mp3", "wav", "ogg"])
 
-        if audio_file is not None:
+        if uploaded_audio:
+            st.audio(uploaded_audio)
             with st.spinner("Processing audio..."):
-                response_text, audio_response = process_audio(audio_file)
-                st.text(f"Response: {response_text}")
-                st.audio(audio_response)
+                response_text, response_audio = process_audio(uploaded_audio)
+                if response_text:
+                    st.markdown(f"**Response Text:** {response_text}")
+                    st.audio(response_audio)
 
 # Run the app
 if __name__ == "__main__":