Update process_energy.py

process_energy.py

@@ -72,7 +72,7 @@ import time
 import numpy as np
 from tqdm import tqdm
 from scipy.io.wavfile import write
-from moviepy.editor import CompositeVideoClip, VideoFileClip, AudioFileClip
+from moviepy.editor import ImageClip, CompositeVideoClip, VideoFileClip, AudioFileClip
 from moviepy.video.fx.all import fadein, fadeout, loop
 from PIL import Image, ImageDraw, ImageFont
 
@@ -167,20 +167,59 @@ def activate_transfer():
     new_mp = recipient_mp + transferred_mp
     return new_mp
 
-# Fungsi untuk membuat video afirmasi dengan audio tanpa teks
-def create_affirmation_video_with_audio(audio_filename, output_filename="affirmation_video.mp4"):
+from PIL import Image, ImageDraw, ImageFont
+from moviepy.editor import ImageClip, CompositeVideoClip, VideoFileClip, AudioFileClip
+from moviepy.video.fx.loop import loop
+import os
+
+# Fungsi untuk membuat frame gambar afirmasi
+def create_affirmation_frame(text, width=640, height=480, duration=5):
+    frame = Image.new("RGBA", (width, height), color=(0, 0, 0, 0))  # Transparan
+    draw = ImageDraw.Draw(frame)
+    
+    font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"  # Path font default
+    font_size = 48
+
+    # Menggunakan font default
+    try:
+        font = ImageFont.truetype(font_path, font_size)
+    except IOError:
+        font = ImageFont.load_default()
+    
+    bbox = draw.textbbox((0, 0), text, font=font)
+    width_text = bbox[2] - bbox[0]
+    height_text = bbox[3] - bbox[1]
+    draw.text(((width - width_text) / 2, (height - height_text) / 2), text, font=font, fill="white")
+
+    return frame
+
+# Fungsi untuk membuat video afirmasi dengan audio
+def create_affirmation_video_with_audio(tujuan, audio_filename, output_filename="affirmation_video.mp4"):
+    # Create frame for the chosen goal
+    frame = create_affirmation_frame(tujuan.upper(), width=640, height=480)
+
+    # Save frame as image
+    image_filename = "frame_tujuan.png"
+    frame.save(image_filename)
+
+    # Load image as video clip
+    clip = ImageClip(image_filename).set_duration(180)  # Durasi teks 3 menit
+
     # Load background GIF and loop it
     background_clip = VideoFileClip("spin_energy.gif").fx(loop, duration=180)
 
     # Load the generated audio file
     audio_clip = AudioFileClip(audio_filename)
 
-    # Set background duration and add audio
-    video = background_clip.set_duration(180).set_audio(audio_clip)
+    # Concatenate text clip with background
+    video = CompositeVideoClip([background_clip, clip]).set_duration(180).set_audio(audio_clip)
 
-    # Write the result to a file with a lower resolution to reduce size
+    # Write the result to a file
     video.write_videofile(output_filename, fps=24, codec='libx264', bitrate='500k')  # Bitrate rendah untuk ukuran file lebih kecil
 
+    # Remove temporary image file
+    os.remove(image_filename)
+
 # Fungsi untuk memproses transfer energi
 def process_energy_transfer(tujuan, nama_anda):
     status = "Mengumpulkan Energi Semesta Digital ..."
@@ -206,7 +245,7 @@ def process_energy_transfer(tujuan, nama_anda):
     
     new_mp = sum(energy_data)
     video_filename = "affirmation_video.mp4"
-    create_affirmation_video_with_audio(filename, video_filename)
+    create_affirmation_video_with_audio(tujuan, filename, video_filename)
     video_path = os.path.abspath(video_filename)
     return (
         f"Transfer Selesai! MP baru Anda: {new_mp:.2f}\n\n"
@@ -217,4 +256,4 @@ def process_energy_transfer(tujuan, nama_anda):
         filename,
         status,
         video_path
-    )
+    )