gemini changed and fool proffing

sentiment_analysis.py

@@ -8,7 +8,10 @@ def get_analysis(file_path)->str:
     """Makes the pipeline to perform the sentiment analysis on a .wav file"""
     transcript = get_transcription_from_sound(file_path)
     
-    analysis = ask_gemini(transcript)
+    if len(transcript) > 0:
+        analysis = ask_gemini(transcript)
+    else :
+        analysis = "Did not provide any message in order to be analysed"
     
     return transcript , analysis
 

streamlit_app.py

@@ -31,8 +31,8 @@ def display_message(role, content, image_path=None):
     """Displays the messages on the screen along with the plots"""
     if role == "user":
         st.chat_message(role).markdown(f"**User:** {content}")
-    elif role == "gemini":
-        st.chat_message(role).markdown(f"**Gemini:** {content}")
+    elif role == "ai":
+        st.chat_message(role).markdown(f"**Ai:** {content}")
         if image_path:
             # Display the image if it exists
             try:
@@ -70,18 +70,18 @@ def main():
         save_audio(audio["bytes"], save_path)
 
         time.sleep(0.1)
-        with st.spinner('Fetching response from Gemini...'):
+        with st.spinner('Fetching response from Ai...'):
             user_message, gemini_response = get_analysis("recorded_audio.wav")
 
         st.session_state.messages.append({"role": "user", "content": user_message})
-        st.session_state.messages.append({"role": "gemini", "content": gemini_response, "image_url": "plot.png"})
+        st.session_state.messages.append({"role": "ai", "content": gemini_response, "image_url": "plot.png"})
 
     # Display the chat history
     for msg in st.session_state.messages:
         # Check if the message is from Gemini and display the image accordingly
         if msg["role"] == "user":
             display_message(msg["role"], msg["content"])
-        elif msg["role"] == "gemini":
+        elif msg["role"] == "ai":
             display_message(msg["role"], msg["content"], msg.get("image_url"))
     
     time.sleep(0.1)