Update sozo_gen.py
Browse files- sozo_gen.py +2 -2
sozo_gen.py
CHANGED
|
@@ -31,13 +31,13 @@ MAX_CHARTS, VIDEO_SCENES = 5, 5
|
|
| 31 |
# --- Gemini API Initialization ---
|
| 32 |
API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 33 |
if not API_KEY:
|
| 34 |
-
raise ValueError("
|
| 35 |
genai.configure(api_key=API_KEY)
|
| 36 |
|
| 37 |
# --- Helper Functions ---
|
| 38 |
def load_dataframe_safely(buf, name: str):
|
| 39 |
ext = Path(name).suffix.lower()
|
| 40 |
-
df = (pd.read_excel if ext in (".xlsx", ".xls") else pd.read_csv)(buf)
|
| 41 |
df.columns = df.columns.astype(str).str.strip()
|
| 42 |
df = df.dropna(how="all")
|
| 43 |
if df.empty or len(df.columns) == 0: raise ValueError("No usable data found")
|
|
|
|
| 31 |
# --- Gemini API Initialization ---
|
| 32 |
API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 33 |
if not API_KEY:
|
| 34 |
+
raise ValueError("GOOGLE_API_KEY environment variable not set.")
|
| 35 |
genai.configure(api_key=API_KEY)
|
| 36 |
|
| 37 |
# --- Helper Functions ---
|
| 38 |
def load_dataframe_safely(buf, name: str):
|
| 39 |
ext = Path(name).suffix.lower()
|
| 40 |
+
df = (pd.read_excel if ext in (".xlsx", ".xls") else pd.read_csv)(buf)
|
| 41 |
df.columns = df.columns.astype(str).str.strip()
|
| 42 |
df = df.dropna(how="all")
|
| 43 |
if df.empty or len(df.columns) == 0: raise ValueError("No usable data found")
|