Spaces:
Running on CPU Upgrade

akhaliq HF staff commited on
Commit
9cc0351
·
1 Parent(s): c6c5393

add open router

Browse files
Files changed (4) hide show
  1. app.py +5 -3
  2. app_openrouter.py +22 -0
  3. pyproject.toml +1 -1
  4. requirements.txt +1 -1
app.py CHANGED
@@ -30,6 +30,7 @@ from app_qwen_coder import demo as demo_qwen_coder
30
  from app_nvidia_coder import demo as demo_nvidia_coder
31
  from app_openai import demo as demo_openai
32
  from app_sambanova_coder import demo as demo_sambanova_coder
 
33
  from utils import get_app
34
  import gradio as gr
35
 
@@ -60,6 +61,7 @@ gr.load(
60
 
61
  # Create mapping of providers to their demos
62
  PROVIDERS = {
 
63
  "OpenAI Coder": demo_openai_coder,
64
  "Sambanova Coder": demo_sambanova_coder,
65
  "Gemini Coder": demo_gemini_coder,
@@ -101,13 +103,13 @@ with demo:
101
 
102
  provider_dropdown = gr.Dropdown(
103
  choices=list(PROVIDERS.keys()),
104
- value="OpenAI Coder",
105
  label="Select code snippet"
106
  )
107
  code_display = gr.Code(
108
  label="Provider Code Snippet",
109
  language="python",
110
- value=PROVIDER_SNIPPETS["OpenAI Coder"]
111
  )
112
 
113
  def update_code(provider):
@@ -122,7 +124,7 @@ with demo:
122
 
123
  selected_demo = get_app(
124
  models=list(PROVIDERS.keys()),
125
- default_model="OpenAI Coder",
126
  src=PROVIDERS,
127
  dropdown_label="Select Provider",
128
  )
 
30
  from app_nvidia_coder import demo as demo_nvidia_coder
31
  from app_openai import demo as demo_openai
32
  from app_sambanova_coder import demo as demo_sambanova_coder
33
+ from app_openrouter import demo as demo_openrouter
34
  from utils import get_app
35
  import gradio as gr
36
 
 
61
 
62
  # Create mapping of providers to their demos
63
  PROVIDERS = {
64
+ "OpenRouter Coder": demo_openrouter,
65
  "OpenAI Coder": demo_openai_coder,
66
  "Sambanova Coder": demo_sambanova_coder,
67
  "Gemini Coder": demo_gemini_coder,
 
103
 
104
  provider_dropdown = gr.Dropdown(
105
  choices=list(PROVIDERS.keys()),
106
+ value="OpenRouter Coder",
107
  label="Select code snippet"
108
  )
109
  code_display = gr.Code(
110
  label="Provider Code Snippet",
111
  language="python",
112
+ value=PROVIDER_SNIPPETS["OpenRouter Coder"]
113
  )
114
 
115
  def update_code(provider):
 
124
 
125
  selected_demo = get_app(
126
  models=list(PROVIDERS.keys()),
127
+ default_model="OpenRouter Coder",
128
  src=PROVIDERS,
129
  dropdown_label="Select Provider",
130
  )
app_openrouter.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import ai_gradio
2
+
3
+ from utils_ai_gradio import get_app
4
+
5
+ # Get the OpenAI models but keep their full names for loading
6
+ OPENROUTER_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("openrouter:")]
7
+
8
+ # Create display names without the prefix
9
+ OPENROUTER_MODELS_DISPLAY = [k.replace("openrouter:", "") for k in OPENROUTER_MODELS_FULL]
10
+
11
+ # Create and launch the interface using get_app utility
12
+ demo = get_app(
13
+ models=OPENROUTER_MODELS_FULL, # Use the full names with prefix
14
+ default_model=OPENROUTER_MODELS_FULL[1],
15
+ dropdown_label="Select OpenRouter Model",
16
+ choices=OPENROUTER_MODELS_DISPLAY, # Display names without prefix
17
+ fill_height=True,
18
+ coder=True,
19
+ )
20
+
21
+ if __name__ == "__main__":
22
+ demo.launch()
pyproject.toml CHANGED
@@ -38,7 +38,7 @@ dependencies = [
38
  "langchain>=0.3.14",
39
  "chromadb>=0.5.23",
40
  "openai>=1.55.0",
41
- "ai-gradio[crewai,deepseek,gemini,groq,hyperbolic,openai,smolagents,transformers, langchain, mistral,minimax,nvidia, qwen]>=0.2.42",
42
  ]
43
 
44
  [tool.uv.sources]
 
38
  "langchain>=0.3.14",
39
  "chromadb>=0.5.23",
40
  "openai>=1.55.0",
41
+ "ai-gradio[crewai,deepseek,gemini,groq,hyperbolic,openai,smolagents,transformers, langchain, mistral,minimax,nvidia, qwen]>=0.2.44",
42
  ]
43
 
44
  [tool.uv.sources]
requirements.txt CHANGED
@@ -2,7 +2,7 @@
2
  # uv pip compile pyproject.toml -o requirements.txt
3
  accelerate==1.2.1
4
  # via ai-gradio
5
- ai-gradio==0.2.42
6
  # via anychat (pyproject.toml)
7
  aiofiles==23.2.1
8
  # via gradio
 
2
  # uv pip compile pyproject.toml -o requirements.txt
3
  accelerate==1.2.1
4
  # via ai-gradio
5
+ ai-gradio==0.2.44
6
  # via anychat (pyproject.toml)
7
  aiofiles==23.2.1
8
  # via gradio