Update src/streamlit_app.py

src/streamlit_app.py

@@ -161,7 +161,6 @@
 
 
 
-
 import streamlit as st
 import requests
 import csv
@@ -173,34 +172,34 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 import os
 import threading
 import schedule
-import os
 
 st.set_page_config(page_title="Temple Dekho WhatsApp Promo Scheduler", layout="wide")
 st.title("📅 Temple Dekho WhatsApp Message Scheduler")
 
-# Create logs directory if not exists
+# Logs directory
 log_dir = "/tmp/logs"
 os.makedirs(log_dir, exist_ok=True)
 
 # --- Helper Functions ---
-def upload_media(file, access_token, phone_number_id):
-    mime_type = mimetypes.guess_type(file.name)[0] or "application/octet-stream"
-    files = {
-        'file': (file.name, file.getvalue(), mime_type),
-        'type': (None, mime_type),
-        'messaging_product': (None, 'whatsapp')
-    }
-    headers = {
-        'Authorization': f'Bearer {access_token}'
-    }
-    url = f"https://graph.facebook.com/v19.0/{phone_number_id}/media"
-
-    try:
-        res = requests.post(url, headers=headers, files=files)
-        res.raise_for_status()
-        return res.json().get("id"), res.status_code, res.text
-    except Exception as e:
-        return None, 500, str(e)
+def upload_media(file_path, access_token, phone_number_id):
+    mime_type = mimetypes.guess_type(file_path)[0] or "application/octet-stream"
+    with open(file_path, "rb") as f:
+        files = {
+            'file': (os.path.basename(file_path), f, mime_type),
+            'type': (None, mime_type),
+            'messaging_product': (None, 'whatsapp')
+        }
+        headers = {
+            'Authorization': f'Bearer {access_token}'
+        }
+        url = f"https://graph.facebook.com/v19.0/{phone_number_id}/media"
+
+        try:
+            res = requests.post(url, headers=headers, files=files)
+            res.raise_for_status()
+            return res.json().get("id"), res.status_code, res.text
+        except Exception as e:
+            return None, 500, str(e)
 
 def send_message(phone, payload, headers, retries=3, backoff_intervals=[5, 15, 30]):
     for attempt in range(retries):
@@ -251,7 +250,7 @@ def execute_campaign(params):
                 logfile.write(result + "\n")
                 logfile.flush()
 
-# --- UI Form to Schedule Campaign ---
+# --- UI Form ---
 with st.form("schedule_form"):
     st.subheader("🗓️ Schedule WhatsApp Campaign")
     col1, col2 = st.columns(2)
@@ -288,27 +287,45 @@ if submitted:
     if not csv_file:
         st.error("CSV file is required.")
     else:
+        # Save CSV to /tmp
+        csv_path = f"/tmp/{csv_file.name}"
+        with open(csv_path, "wb") as f:
+            f.write(csv_file.getbuffer())
+
         media_id = existing_media_id.strip() if existing_media_id.strip() else None
+
+        # Save media file to /tmp
         if media_file and not media_id:
-            media_id, status_code, response = upload_media(media_file, access_token, phone_number_id)
+            media_path = f"/tmp/{media_file.name}"
+            with open(media_path, "wb") as f:
+                f.write(media_file.getbuffer())
+            media_id, status_code, response = upload_media(media_path, access_token, phone_number_id)
             if not media_id:
                 st.error(f"Media upload failed. Status: {status_code}\nResponse: {response}")
+                st.stop()
 
-        csv_file.seek(0)
-        decoded = csv_file.read().decode('utf-8').strip()
-        rows = csv.reader(io.StringIO(decoded))
-        next(rows, None)
-
+        # Process CSV
         phones = []
-        for row in rows:
-            if not row or not row[0].strip():
-                continue
-            num = row[0].strip()
-            if num.isdigit():
-                if len(num) == 10:
-                    phones.append("91" + num)
-                elif len(num) == 12 and num.startswith("91"):
-                    phones.append(num)
+        try:
+            with open(csv_path, "r", encoding="utf-8-sig") as f:
+                reader = csv.reader(f)
+                next(reader, None)
+                for row in reader:
+                    if not row or not row[0].strip():
+                        continue
+                    num = row[0].strip()
+                    if num.isdigit():
+                        if len(num) == 10:
+                            phones.append("91" + num)
+                        elif len(num) == 12 and num.startswith("91"):
+                            phones.append(num)
+        except Exception as e:
+            st.error(f"Error reading CSV: {e}")
+            st.stop()
+
+        if not phones:
+            st.error("No valid phone numbers found in CSV.")
+            st.stop()
 
         headers = {
             "Authorization": f"Bearer {access_token}",
@@ -340,7 +357,7 @@ if submitted:
                 })
             components += buttons
         except:
-            pass
+            pass  # ignore if button input is malformed
 
         scheduled_datetime = datetime.combine(schedule_date, schedule_time)
 
@@ -367,5 +384,3 @@ if submitted:
         threading.Thread(target=run_scheduler, daemon=True).start()
 
         st.success(f"✅ Campaign scheduled for {scheduled_datetime.strftime('%Y-%m-%d %H:%M:%S')}. UI can now be closed.")
-
-