Replace Gemini with OpenAI

app.py

@@ -2,11 +2,12 @@ import pandas as pd
 import gradio as gr
 import numpy as np
 import os
-import google.generativeai as genai  # requires GOOGLE_API_KEY set as env var
+import openai  # requires OPENAI_API_KEY set as env var
+
+# 1. Configure OpenAI
+openai.api_key = os.getenv("OPENAI_API_KEY")
+MODEL_NAME = "gpt-5"
 
-# 1. Configure Gemini
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-model = genai.GenerativeModel("gemini-2.5-flash")
 
 # 2. Load Excel data
 # Load all sheets
@@ -177,9 +178,17 @@ Rules:
     """
 
     try:
-        # Ask Gemini
-        response = model.generate_content(prompt)
-        code = response.text.strip()
+        # Ask OpenAI
+        response = openai.ChatCompletion.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0.2,
+            max_tokens=1200
+        )
+        code = response.choices[0].message["content"].strip()
 
         # Strip markdown fences if any
         if code.startswith("```"):