Update app.py

app.py

@@ -9,7 +9,10 @@ from transformers import pipeline
 
 API_KEY = os.getenv("API_KEY")
 
-API_URL = "https://api-inference.huggingface.co/models/dima806/facial_emotions_image_detection"
+IMAGE_API_URL = "https://api-inference.huggingface.co/models/dima806/facial_emotions_image_detection"
+headers = {"Authorization": "Bearer "+ API_KEY+""}
+
+EMOTIONS_API_URL = "https://api-inference.huggingface.co/models/SamLowe/roberta-base-go_emotions"
 headers = {"Authorization": "Bearer "+ API_KEY+""}
 
 reader = easyocr.Reader(['en'], gpu=False)
@@ -22,7 +25,7 @@ def query(image):
     _, buffer = cv2.imencode('.jpg', image_data)
     binary_data = buffer.tobytes()
 
-    response = requests.post(API_URL, headers=headers, data=binary_data)
+    response = requests.post(IMAGE_API_URL, headers=headers, data=binary_data)
     result = {item['label']: item['score'] for item in response.json()}
 
     return result
@@ -42,12 +45,11 @@ def text_extraction(image):
     return image, text_content, facial_data
 
 def analyze_sentiment(text):
-    results = sentiment_analysis(text)
-    sentiment_results = {result['label']: result['score'] for result in results}
+    results = requests.post(EMOTIONS_API_URL, headers=headers, json=text)
+    sentiment_results = {result['label']: result['score'] for result in results.json()}
     return sentiment_results
 
 def get_sentiment_emoji(sentiment):
-    # Define the emojis corresponding to each sentiment
     emoji_mapping = {
         "disappointment": "😞",
         "sadness": "😢",
@@ -182,15 +184,17 @@ with block:
                     gr.Textbox(label="Text Content")
     
                     output_text_sentiment = gr.Textbox(label="Text Sentiment")
-                
-                audio = gr.Audio(label="Input Audio", show_label=False, type="filepath")
-                
-                with gr.Row():
-                    sentiment_option = gr.Radio(choices=["Sentiment Only", "Sentiment + Score"], label="Select an option")
+
+                with gr.Blocks():
+                    with gr.Row():
+                        audio = gr.Audio(label="Input Audio", show_label=False, type="filepath")
                     
-                    lang_str = gr.Textbox(label="Language")
-                    text = gr.Textbox(label="Transcription")
-                    sentiment_output = gr.Textbox(label="Audio Text Sentiment")
+                        with gr.Row():
+                            sentiment_option = gr.Radio(choices=["Sentiment Only", "Sentiment + Score"], label="Select an option")
+                            
+                            lang_str = gr.Textbox(label="Language")
+                            text = gr.Textbox(label="Transcription")
+                            sentiment_output = gr.Textbox(label="Audio Text Sentiment")
 
         
         btn = gr.Button("Run")