update

app.py

@@ -1,19 +1,20 @@
 import os
 import requests
+import openai
 import streamlit as st
-from transformers import pipeline
 from PIL import Image
 from dotenv import load_dotenv
-import time
 
 # Load environment variables from .env file
 load_dotenv()
 
-# Set up the Hugging Face API URLs and your API key
+# Set up the Hugging Face API URL and your API key
 emotion_model_url = "https://api-inference.huggingface.co/models/trpakov/vit-face-expression"
-text_model_url = "https://api-inference.huggingface.co/models/mrm8488/t5-base-finetuned-emotion"
 headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_KEY')}"}
 
+# Set up OpenAI API key
+openai.api_key = os.getenv('OPENAI_API_KEY')
+
 # Function to query the facial expression recognition model
 def query_emotion(filename):
     with open(filename, "rb") as f:
@@ -26,33 +27,27 @@ def query_emotion(filename):
         st.error("Error detecting facial expression: " + response.text)
         return None
 
-# Function to generate a joke or uplifting text based on the mood
+# Function to generate a joke or motivational text using OpenAI
 def generate_text_based_on_mood(emotion):
     try:
-        prompt = f"Generate a light-hearted joke or uplifting message for someone who is feeling {emotion}."
-        
-        # Retry mechanism for text generation
-        for attempt in range(5):  # Retry up to 5 times
-            response = requests.post(text_model_url, headers=headers, json={"inputs": prompt})
-            if response.status_code == 200:
-                generated_text = response.json()[0]['generated_text']
-                if generated_text.strip():  # Ensure the response is not empty
-                    return generated_text
-                else:
-                    st.warning("Received an empty joke, retrying...")
-            elif response.status_code == 503:  # Service Unavailable
-                st.warning("Model is loading, retrying...")
-                time.sleep(5)  # Wait before retrying
-            else:
-                st.error("Error generating text: " + response.text)
-                return "Sorry, I couldn't come up with a joke at this moment."
-        
-        st.error("Failed to generate text after multiple attempts.")
-        return "Sorry, I couldn't come up with a joke at this moment."
+        # Create a prompt for OpenAI
+        prompt = f"Generate a light-hearted joke or a motivational quote for someone who is feeling {emotion}."
         
+        # Call OpenAI's API
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Use the desired model
+            messages=[
+                {"role": "user", "content": prompt}
+            ]
+        )
+
+        # Extract the generated text
+        generated_text = response.choices[0].message['content']
+        return generated_text.strip()
+
     except Exception as e:
         st.error(f"Error generating text: {e}")
-        return "Sorry, I couldn't come up with a joke at this moment."
+        return "Sorry, I couldn't come up with a joke or motivation at this moment."
 
 # Function to convert text to speech using gTTS
 def text_to_speech(text):