marks commited on
Commit
25262d5
·
1 Parent(s): cb5c924

Fixed AI models grabbing ids

Browse files
Files changed (2) hide show
  1. app.py +4 -6
  2. models.py +2 -6
app.py CHANGED
@@ -66,17 +66,15 @@ class PodcasterUI:
66
  with gr.Column():
67
  openrouter_model = gr.Dropdown(
68
  label='AI Model',
69
- choices=[(name, id) for id, name in self.models], # Swap order for display
70
- value=self.models[0][1] if len(self.models) > 1 else None,
71
- type="index" # Use index to get the second element (id) from tuple
72
  )
73
 
74
  with gr.Column():
75
  voice_model = gr.Dropdown(
76
  label='Voice',
77
- choices=[(name, id) for id, name in self.voices], # Swap order for display
78
- value=self.voices[0][1] if len(self.voices) > 1 else None,
79
- type="index" # Use index to get the second element (id) from tuple
80
  )
81
 
82
  prompt_input = gr.Textbox(
 
66
  with gr.Column():
67
  openrouter_model = gr.Dropdown(
68
  label='AI Model',
69
+ choices=[(id, name) for id, name in self.models],
70
+ value=self.models[0][0] if len(self.models) > 1 else None,
 
71
  )
72
 
73
  with gr.Column():
74
  voice_model = gr.Dropdown(
75
  label='Voice',
76
+ choices=[(id, name) for id, name in self.voices],
77
+ value=self.voices[0][0] if len(self.voices) > 1 else None,
 
78
  )
79
 
80
  prompt_input = gr.Textbox(
models.py CHANGED
@@ -9,15 +9,11 @@ class OpenRouterRequest(BaseModel):
9
  model: str
10
  messages: List[Message]
11
 
12
- class Choice(BaseModel):
13
  message: Message
14
- index: int = 0
15
- finish_reason: Optional[str] = None
16
 
17
  class OpenRouterResponse(BaseModel):
18
- id: str
19
- choices: List[Choice]
20
- model: str
21
 
22
  class OpenRouterModel(BaseModel):
23
  id: str
 
9
  model: str
10
  messages: List[Message]
11
 
12
+ class OpenRouterChoice(BaseModel):
13
  message: Message
 
 
14
 
15
  class OpenRouterResponse(BaseModel):
16
+ choices: List[OpenRouterChoice]
 
 
17
 
18
  class OpenRouterModel(BaseModel):
19
  id: str