Update app.py

app.py

@@ -40,12 +40,15 @@ def delay_execution_10(pagent, **kwargs) -> bool:
 class MyAgent:
     def __init__(self):
         gemini_api_key = os.getenv("GEMINI_API_KEY")
+        openai_api_key = os.getenv("OPENAI_API_KEY")
         if not gemini_api_key:
             raise ValueError("GEMINI_API_KEY not set in environment variables.")
         
         self.model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash-lite",
-            api_key=gemini_api_key,
+            # model_id="gemini/gemini-2.0-flash-lite",
+            model_id="openai/gpt-4o",
+            # api_key=gemini_api_key,
+            api_key=openai_api_key,
             system_prompt=SYSTEM_PROMPT
         )