Update app.py

app.py

@@ -4,7 +4,11 @@ import json
 import gradio as gr
 from openai import OpenAI
 
-client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+#client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+client = OpenAI(
+    api_key=os.environ["OPENROUTER_API_KEY"],
+    base_url="https://openrouter.ai/api/v1", 
+)  # for using openrouter instead
 
 BASIC_PROMPT = """You are a helpful assistant that processes emails."""
 
@@ -304,7 +308,8 @@ def run_agent(system_prompt, email, show_raw_output):
 
     try:
         first_response = client.chat.completions.create(
-            model="gpt-4o-mini",
+            #model="gpt-4o-mini",
+            model="openai/gpt-4o-mini", # for using openrouter instead
             messages=[
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": email},
@@ -326,7 +331,8 @@ def run_agent(system_prompt, email, show_raw_output):
             )
 
             second_response = client.chat.completions.create(
-                model="gpt-4o-mini",
+                #model="gpt-4o-mini",
+                model="openai/gpt-4o-mini", # for using openrouter instead
                 messages=[
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": email},