Update app.py

app.py

@@ -12,222 +12,210 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import gradio as gr
+from groq import Groq
 import numpy as np
 import soundfile as sf
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
+import matplotlib.pyplot as plt
+from PIL import Image
+import io
 
-# Load the pre-trained Sentence Transformer model
-model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
+# Initialize Groq client
+client = Groq(
+    api_key=os.environ.get("GROQ_API_KEY"),
+)
 
-# Function to generate a subliminal message
-def generate_subliminal(message, sample_rate=44100, duration=30, subliminal_volume=0.1):
-    t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
-    
-    # Convert the message to a series of frequencies
-    frequencies = [ord(char) * 10 for char in message]  # Simple mapping of characters to frequencies
-    
-    subliminal_wave = np.zeros_like(t)
-    for freq in frequencies:
-        subliminal_wave += np.sin(2 * np.pi * freq * t)
-    
-    # Normalize and adjust volume
-    subliminal_wave = subliminal_wave / np.max(np.abs(subliminal_wave)) * subliminal_volume
-    
-    return subliminal_wave
-
-# Function to combine subliminal message with a frequency tone
-def combine_audio(subliminal_waveform, frequency, duration=30, sample_rate=44100):
-    t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
-    sine_wave = 0.5 * np.sin(2 * np.pi * frequency * t)
-    combined_waveform = sine_wave + subliminal_waveform
-    
-    # Normalize the combined waveform
-    combined_waveform = combined_waveform / np.max(np.abs(combined_waveform))
-    
-    file_name = "combined_audio.wav"
-    sf.write(file_name, combined_waveform, sample_rate)
-    return file_name
+# Load pre-trained sentence transformer model
+model = SentenceTransformer('all-MiniLM-L6-v2')
 
-# Define vibrational frequencies and corresponding advice
+# Vibrational advice dictionary
 vibrational_advice = {
     "anxious": {
-        "name": "396 Hz - Liberation from Fear",
         "frequency": 396,
-        "advice": "1. Practice deep breathing exercises to calm your mind.\n"
-                  "2. Write down your fears and reflect on their source.\n"
-                  "3. Surround yourself with positive affirmations and energy.\n"
-                  "4. Visualize letting go of fear during meditation.\n"
-                  "5. Listen to 396 Hz music while meditating.",
-        "subliminal": "You are calm and at peace."
+        "name": "Liberating Guilt and Fear",
+        "advice": "The 396 Hz frequency is associated with releasing fear and guilt. Try these techniques:\n"
+                  "1. Practice deep, slow breathing exercises for 5-10 minutes.\n"
+                  "2. Use positive affirmations like 'I am safe' or 'I release my fears'.\n"
+                  "3. Ground yourself by walking barefoot on grass or soil.\n"
+                  "4. Listen to 396 Hz music while meditating or before sleep.\n"
+                  "5. Try progressive muscle relaxation to release physical tension."
     },
     "stressed": {
-        "name": "417 Hz - Facilitating Change",
-        "frequency": 417,
-        "advice": "1. Take short breaks to clear your mind during work.\n"
-                  "2. Practice progressive muscle relaxation to reduce tension.\n"
-                  "3. Break down overwhelming tasks into smaller steps.\n"
-                  "4. Focus on things you can control and let go of what you can't.\n"
-                  "5. Listen to 417 Hz music to release negative energy.",
-        "subliminal": "You handle challenges with ease."
+        "frequency": 528,
+        "name": "Transformation and Miracles",
+        "advice": "The 528 Hz frequency is known as the 'miracle tone' for transformation and DNA repair. Consider these practices:\n"
+                  "1. Start your day with a gratitude journal, listing 3-5 things you're thankful for.\n"
+                  "2. Practice mindfulness meditation focusing on your breath for 10-15 minutes.\n"
+                  "3. Take regular breaks during work to stretch and reset your mind.\n"
+                  "4. Listen to 528 Hz music during your self-care routine or while working.\n"
+                  "5. Engage in creative activities like drawing, coloring, or crafting to reduce stress."
     },
     "calm": {
-        "name": "528 Hz - Transformation and Miracles",
-        "frequency": 528,
-        "advice": "1. Embrace gratitude by listing things you're thankful for.\n"
-                  "2. Focus on activities that enhance your inner peace.\n"
-                  "3. Connect with nature to maintain your calm state.\n"
-                  "4. Visualize your energy expanding harmoniously.\n"
-                  "5. Listen to 528 Hz music to amplify feelings of love and peace.",
-        "subliminal": "You radiate peace and harmony."
+        "frequency": 741,
+        "name": "Awakening Intuition",
+        "advice": "The 741 Hz frequency is associated with awakening intuition and solving problems. Enhance your state with these tips:\n"
+                  "1. Start a dream journal to tap into your subconscious mind.\n"
+                  "2. Practice visualization techniques, imagining your goals and aspirations.\n"
+                  "3. Engage in mindful movement like yoga or tai chi.\n"
+                  "4. Spend time in nature to maintain your sense of calm and clarity.\n"
+                  "5. Listen to 741 Hz music during problem-solving or creative sessions."
     },
     "sad": {
-        "name": "639 Hz - Harmonizing Relationships",
-        "frequency": 639,
-        "advice": "1. Reach out to a friend or loved one to talk about your feelings.\n"
-                  "2. Practice self-compassion and avoid self-criticism.\n"
-                  "3. Engage in creative expression, such as journaling or drawing.\n"
-                  "4. Focus on rebuilding positive relationships in your life.\n"
-                  "5. Listen to 639 Hz music to nurture emotional healing.",
-        "subliminal": "You are worthy of love and happiness."
+        "frequency": 417,
+        "name": "Facilitating Change",
+        "advice": "The 417 Hz frequency is believed to facilitate change and let go of negative energy. Try these approaches:\n"
+                  "1. Write a letter expressing your feelings, then safely burn or discard it as a symbolic release.\n"
+                  "2. Engage in physical exercise to boost endorphins and improve mood.\n"
+                  "3. Reach out to a friend or loved one for support and connection.\n"
+                  "4. Practice self-compassion meditation, treating yourself with kindness.\n"
+                  "5. Listen to 417 Hz music while journaling or during a relaxing bath."
     },
     "angry": {
-        "name": "741 Hz - Expression and Solutions",
-        "frequency": 741,
-        "advice": "1. Take a step back and breathe deeply to cool down.\n"
-                  "2. Channel your anger into physical activity, like a workout.\n"
-                  "3. Practice mindfulness to understand the root of your anger.\n"
-                  "4. Use 'I' statements when expressing your feelings to others.\n"
-                  "5. Listen to 741 Hz music to balance your energy.",
-        "subliminal": "You respond to life with patience and understanding."
+        "frequency": 852,
+        "name": "Returning to Spiritual Order",
+        "advice": "The 852 Hz frequency is associated with returning to spiritual order and inner strength. Consider these techniques:\n"
+                  "1. Practice 'square breathing': inhale for 4 counts, hold for 4, exhale for 4, hold for 4, repeat.\n"
+                  "2. Use positive self-talk to reframe the situation causing anger.\n"
+                  "3. Engage in a physical activity like running or punching a pillow to release tension.\n"
+                  "4. Write down your angry thoughts, then rewrite them from a calm perspective.\n"
+                  "5. Listen to 852 Hz music during meditation or while practicing forgiveness exercises."
     },
     "fearful": {
-        "name": "852 Hz - Awakening Intuition",
-        "frequency": 852,
-        "advice": "1. Ground yourself by focusing on the present moment.\n"
-                  "2. Visualize releasing fear and embracing trust.\n"
-                  "3. Focus on building resilience and inner strength.\n"
-                  "4. Journal your fears and look for patterns.\n"
-                  "5. Listen to 852 Hz music to activate your intuition.",
-        "subliminal": "You are safe and protected."
+        "frequency": 639,
+        "name": "Connecting Relationships",
+        "advice": "The 639 Hz frequency is linked to connecting relationships and understanding. When feeling fearful, try these:\n"
+                  "1. Reach out to a trusted friend or family member to share your fears.\n"
+                  "2. Practice 'loving-kindness' meditation to foster feelings of safety and connection.\n"
+                  "3. Create a 'fear ladder' and gradually expose yourself to what you fear in small steps.\n"
+                  "4. Use positive visualization to imagine successfully overcoming your fears.\n"
+                  "5. Listen to 639 Hz music while working on building your support network."
     },
     "confused": {
-        "name": "285 Hz - Returning to Source",
         "frequency": 285,
-        "advice": "1. Take time to reflect in a quiet space.\n"
-                  "2. Ask yourself what truly matters in your current situation.\n"
-                  "3. Break down complex problems into simpler parts.\n"
-                  "4. Meditate to clear your mind and reduce mental clutter.\n"
-                  "5. Listen to 285 Hz music while reflecting or problem-solving.",
-        "subliminal": "Your mind is clear and focused."
+        "name": "Quantum Cognition",
+        "advice": "The 285 Hz frequency is believed to influence energy fields and aid in healing. When feeling confused, consider:\n"
+                  "1. Create a mind map to visually organize your thoughts and ideas.\n"
+                  "2. Take a short walk to clear your mind and gain new perspectives.\n"
+                  "3. Practice the 'brain dump' technique: write down all your thoughts without judgment.\n"
+                  "4. Use the Pomodoro technique to break tasks into manageable chunks.\n"
+                  "5. Listen to 285 Hz music while studying or problem-solving to enhance focus."
+    },
+    "happy": {
+        "frequency": 432,
+        "name": "Harmonizing Vibrations",
+        "advice": "The 432 Hz frequency is associated with harmonizing vibrations and promoting wellbeing. Enhance your happy state with these practices:\n"
+                  "1. Express gratitude by sharing your joy with others or writing in a gratitude journal.\n"
+                  "2. Engage in activities that bring you joy and fulfillment.\n"
+                  "3. Practice 'savoring' by fully immersing yourself in positive experiences.\n"
+                  "4. Share your positive energy through acts of kindness or volunteering.\n"
+                  "5. Listen to 432 Hz music to maintain and amplify your positive vibrations."
     }
 }
 
-# Function to analyze user's emotional state and provide a tone
-def analyze_emotion(input_text, uploaded_file=None):
-    input_embedding = model.encode([input_text])
-    emotions = {
-        "anxious": ["I feel nervous and uneasy.", "I'm afraid of something going wrong."],
-        "stressed": ["I'm overwhelmed with work.", "I feel like I'm under a lot of pressure."],
-        "calm": ["I feel peaceful and at ease.", "Everything is going well."],
-        "sad": ["I feel down and unhappy.", "I've been feeling quite low."],
-        "angry": ["I'm furious about something.", "I can't control my anger."],
-        "fearful": ["I'm scared and worried.", "I have a lot of fear right now."],
-        "confused": ["I don't know what to do.", "I'm feeling uncertain and lost."]
-    }
-    emotion_embeddings = {emotion: model.encode(phrases) for emotion, phrases in emotions.items()}
-    best_emotion = max(emotion_embeddings.items(), key=lambda x: np.max(cosine_similarity(input_embedding, x[1])))[0]
-    advice = vibrational_advice[best_emotion]
-    subliminal_message = advice["subliminal"]
-    subliminal_waveform = generate_subliminal(subliminal_message, duration=30)
-    combined_audio = combine_audio(subliminal_waveform, advice["frequency"], duration=30)
-
-    if uploaded_file:
-        uploaded_audio, sample_rate = sf.read(uploaded_file.name)
-        combined_waveform, _ = sf.read(combined_audio)
-        
-        # Adjust the length of the combined waveform to match the uploaded audio
-        if len(combined_waveform) > len(uploaded_audio):
-            combined_waveform = combined_waveform[:len(uploaded_audio)]
-        else:
-            combined_waveform = np.pad(combined_waveform, (0, len(uploaded_audio) - len(combined_waveform)))
+# Create embeddings for predefined emotions
+emotion_embeddings = {emotion: model.encode([emotion]) for emotion in vibrational_advice.keys()}
+
+def get_closest_emotion(detected_emotion):
+    detected_embedding = model.encode([detected_emotion])
+    similarities = {emotion: cosine_similarity(detected_embedding, embedding)[0][0] 
+                    for emotion, embedding in emotion_embeddings.items()}
+    return max(similarities, key=similarities.get)
+
+def generate_tone(frequency, duration=5, sample_rate=44100):
+    t = np.linspace(0, duration, int(sample_rate * duration), False)
+    tone = np.sin(2 * np.pi * frequency * t)
+    return (tone * 32767).astype(np.int16)
+
+def save_tone(frequency):
+    tone = generate_tone(frequency)
+    filename = f"/tmp/tone_{frequency}Hz.wav"
+    sf.write(filename, tone, 44100)
+    return filename
+
+def generate_waveform(frequency):
+    duration = 0.1  # Short duration for visualization
+    sample_rate = 44100
+    t = np.linspace(0, duration, int(sample_rate * duration), False)
+    tone = np.sin(2 * np.pi * frequency * t)
+    
+    plt.figure(figsize=(10, 4))
+    plt.plot(t, tone)
+    plt.title(f"Waveform of {frequency} Hz Tone")
+    plt.xlabel("Time (s)")
+    plt.ylabel("Amplitude")
+    plt.ylim(-1.1, 1.1)
+    plt.grid(True)
+    
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    plt.close()
+    
+    return Image.open(buf)
+
+def transcribe_audio(audio_path):
+    try:
+        # Open and read the audio file
+        with open(audio_path, "rb") as audio_file:
+            audio_data = audio_file.read()
+
+        # Transcribe the audio using Distil-Whisper
+        transcription = client.audio.transcriptions.create(
+            file=(os.path.basename(audio_path), audio_data),
+            model="distil-whisper-large-v3-en",
+            response_format="verbose_json",
+        )
         
-        combined_waveform = combined_waveform + uploaded_audio
-        combined_audio = "combined_with_upload.wav"
-        sf.write(combined_audio, combined_waveform, sample_rate)
+        return transcription.text
+    except Exception as e:
+        return f"Error in transcription: {str(e)}"
 
-    return best_emotion, advice["name"], advice["advice"], combined_audio
+def vibrational_wellness(input_text, audio_input):
+    if audio_input:
+        input_text = transcribe_audio(audio_input)
+    
+    if not input_text.strip():
+        return "Please enter a description of your emotional state or provide an audio input.", None, None, None
 
-# Gradio interface function
-def wellness_analyzer(input_text, uploaded_file=None):
-    emotion, tone_name, advice, audio_file = analyze_emotion(input_text, uploaded_file)
+    prompt = f"""Analyze the following text and determine the primary emotion expressed. 
+    Choose the most appropriate emotion from this list: anxious, stressed, calm, sad, angry, fearful, confused, happy.\n
+    Input: "{input_text}"\n
+    Emotion:"""
     
-    output = """
-<div style="font-family: Arial, sans-serif; line-height: 1.6;">
-    <h3 style="color: #4a4a4a;">Detected Emotional State: {0}</h3>
-    <p><strong>Recommended Tone:</strong> {1}</p>
-    <p><strong>Personalized Advice:</strong></p>
-    <ol>
-        {2}
-    </ol>
-    <p><em>Listen to the generated audio file for a custom 30-second vibrational experience with subliminal messaging.</em></p>
-</div>
-    """.format(
-        emotion.capitalize(),
-        tone_name,
-        "".join("<li>{0}</li>".format(advice_item.strip()) for advice_item in advice.split("\n") if advice_item.strip())
-    )
+    detected_emotion = client.completions.create(
+        model="gpt-3.5-turbo",
+        prompt=prompt,
+        max_tokens=10,
+        temperature=0,
+    ).choices[0].text.strip()
+
+    closest_emotion = get_closest_emotion(detected_emotion)
+    frequency = vibrational_advice[closest_emotion]["frequency"]
     
-    return output, audio_file
-
-# Define custom CSS
-custom_css = """
-.footer {
-    text-align: center;
-    margin-top: 20px;
-    font-style: italic;
-}
-"""
-
-# Update the Gradio interface
-with gr.Blocks(css=custom_css) as interface:
-    gr.Markdown(
-        """
-        <div style="text-align: center; max-width: 800px; margin: 0 auto;">
-            <h1>Vers3Dynamics Vibrational Wellness 🌟</h1>
-            <p>Discover your emotional resonance and receive personalized vibrational advice with subliminal messaging.</p>
-        </div>
-        """
-    )
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            input_text = gr.Textbox(
-                label="Describe your current emotional state 🤲",
-                placeholder="How are you feeling right now?",
-                lines=3
-            )
-            uploaded_file = gr.File(label="Upload music (optional) 🎵")
-            analyze_btn = gr.Button("Analyze Mood", variant="primary")
-
-        with gr.Column(scale=3):
-            output_text = gr.HTML(label="Analysis and Advice")
-            output_audio = gr.Audio(label="Your Custom 30-second Vibrational Tone with Subliminal Message 🎶", type="filepath")
-
-    gr.Markdown(
-        """
-        <div class="footer">
-            <p>Vers3Dynamics - Harmonizing your well-being through vibrational frequencies and subliminal messaging.</p>
-        </div>
-        """
-    )
-
-    analyze_btn.click(
-        wellness_analyzer,
-        inputs=[input_text, uploaded_file],
-        outputs=[output_text, output_audio]
-    )
-
-# Launch the interface
-interface.launch()
+    tone_file = save_tone(frequency)
+    waveform_image = generate_waveform(frequency)
+
+    return (vibrational_advice[closest_emotion]["advice"], tone_file, waveform_image)
+
+def launch_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("<h1>Vibrational Wellness Advisor</h1>")
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label="Describe your emotional state", placeholder="E.g., I feel anxious...")
+                audio_input = gr.Audio(label="Or upload an audio file", type="filepath", source="upload")
+                submit_btn = gr.Button("Get Vibrational Advice")
+            with gr.Column():
+                advice_output = gr.Markdown(label="Vibrational Advice")
+                tone_output = gr.Audio(label="Generated Tone")
+                waveform_output = gr.Image(label="Waveform")
+        
+        submit_btn.click(vibrational_wellness, inputs=[input_text, audio_input], outputs=[advice_output, tone_output, waveform_output])
+    
+    demo.launch()
+
+if __name__ == "__main__":
+    launch_interface()