Update app.py

app.py

@@ -7,7 +7,6 @@ import time
 
 
 st.set_page_config(page_title="Exness Tempering App by Adil Khan")
-cover = cv2.imread("cover.png", cv2.IMREAD_UNCHANGED)
 
 def overlay_image(background, overlay):
     background_resized = cv2.resize(background, (overlay.shape[1], overlay.shape[0]))
@@ -18,10 +17,16 @@ def overlay_image(background, overlay):
 
 def image_tempering_app():
     st.title("Exness Image Tempering App")
+    cover_option = st.selectbox("Select Image Type:", ["With Banner", "Without Banner"], index=0)
+    if cover_option == "Without Banner":
+        cover_value = "cover.png"
+    else:
+        cover_value = "cover2.png"
     screenshot_image = st.file_uploader("Upload Screenshot Image", type=["jpg", "jpeg", "png"])
     if screenshot_image:
         file_bytes = np.asarray(bytearray(screenshot_image.read()), dtype=np.uint8)
-        image = cv2.imdecode(file_bytes, cv2.IMREAD_UNCHANGED)
+        image = cv2.imdecode(file_bytes, cv2.IMREAD_UNCHANGED)        
+        cover = cv2.imread(cover_value, cv2.IMREAD_UNCHANGED)
         
         with st.spinner("Processing Image..."):
             result = overlay_image(image, cover)
@@ -31,13 +36,14 @@ def image_tempering_app():
         
         st.download_button("Download Processed Image", data=open(temp_filename, "rb").read(), file_name="processed_image.png", mime="image/png")
 
-def process_video(video_path, output_path):
+def process_video(video_path, output_path, cover_value):
     cap = cv2.VideoCapture(video_path)
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     fps = int(cap.get(cv2.CAP_PROP_FPS))
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    cover = cv2.imread(cover_value, cv2.IMREAD_UNCHANGED)
     out = cv2.VideoWriter(output_path, fourcc, fps, (cover.shape[1], cover.shape[0]))
     
     progress_bar = st.progress(0)
@@ -69,6 +75,11 @@ def process_video(video_path, output_path):
 
 def video_tempering_app():
     st.title("Exness Video Tempering App")
+    cover_option = st.selectbox("Select Video Type:", ["With Banner", "Without Banner"], index=0)
+    if cover_option == "Without Banner":
+        cover_value = "cover.png"
+    else:
+        cover_value = "cover2.png"
     screenshot_video = st.file_uploader("Upload Screen Recording", type=["mp4"])
     if screenshot_video:
         temp_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
@@ -76,7 +87,7 @@ def video_tempering_app():
         output_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
         
         with st.spinner("Processing video..."):
-            process_video(temp_video.name, output_video)
+            process_video(temp_video.name, output_video, cover_value)
         
         st.download_button("Download Processed Video", data=open(output_video, "rb").read(), file_name="processed_video.mp4", mime="video/mp4")