add openrouter api

app.py

@@ -2,12 +2,15 @@ import gradio as gr
 import yt_dlp
 import os
 from openai import OpenAI
-import anthropic
 
 client = OpenAI(
     api_key=os.getenv("OPENAI_API_KEY"),
 )
-claude_client = anthropic.Anthropic()
+
+opnerouter_client =  OpenAI(
+  base_url="https://openrouter.ai/api/v1",
+  api_key=os.getenv("OPENROUTER_API_KEY"),
+)
 
 ydl_opts = {
     'outtmpl': 'demo.m4a',
@@ -41,16 +44,13 @@ def download_audio(url):
 def summarize_text():
     prompt = f"Please summarize the following article in 5 sentences or less use the language of the audio: [{text}]"
     print(prompt)
-    message = claude_client.messages.create(
-        model="claude-3-haiku-20240307",
-        max_tokens=1000,
-        temperature=0.0,
-        # system="Respond only in mandarin",
+    completion = opnerouter_client.chat.completions.create(
+        model="google/gemini-pro",
         messages=[
             {"role": "user", "content": prompt}
         ]
     )
-    return message.content[0].text
+    return completion.choices[0].message.content
 
 with gr.Blocks() as demo:
     with gr.Column():

demo.py

@@ -0,0 +1,19 @@
+from openai import OpenAI
+from os import getenv
+
+# gets API Key from environment variable OPENAI_API_KEY
+client = OpenAI(
+  base_url="https://openrouter.ai/api/v1",
+  api_key=getenv(""),
+)
+
+completion = client.chat.completions.create(
+  model="google/gemini-pro",
+  messages=[
+    {
+      "role": "user",
+      "content": "how to learn rust programming language",
+    },
+  ],
+)
+print(completion.choices[0].message.content)