import matplotlib.pyplot as plt
from transcript_processor import query, get_youtube_transcript

def fetch_and_plot_emotions(youtube_id):
		"""
		Fetches the emotion scores from a YouTube video transcript and plots the data.

		Args:
				youtube_id (str): The ID of the YouTube video.

		Returns:
				str: Path to the saved plot image.
		"""
		try:
				transcript = get_youtube_transcript(youtube_id)
				if "error" in transcript:
						return transcript["error"]
				
				emotion_scores = query({"inputs": transcript})
				print(emotion_scores)
				
				if "error" in emotion_scores:
						return emotion_scores["error"]
				
				return plot_emotion_scores(emotion_scores)
		except Exception as e:
				return str(e)

def plot_emotion_scores(data):
		"""
		Plots emotion scores from the provided JSON data.

		Args:
				data (list): A list of dictionaries containing emotion labels and scores.

		Returns:
				str: Path to the saved plot image.
		"""
		try:
				labels = [item['label'] for item in data[0]]
				scores = [item['score'] for item in data[0]]
				
				plt.figure(figsize=(10, 8))
				plt.barh(labels, scores, color='skyblue')
				plt.xlabel('Score')
				plt.title('Emotion Scores')
				plt.gca().invert_yaxis()  # Highest scores at the top
				plt.tight_layout()
				
				# Save the plot to a file
				plot_path = 'emotion_scores.png'
				plt.savefig(plot_path)
				plt.close()
				
				return plot_path
		except KeyError as e:
				return f"Missing expected key in data: {e}"
		except Exception as e:
				return str(e)