Spaces:
Running
Running
File size: 936 Bytes
7d6d833 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from pydantic import BaseModel
# Define the request and response schema
class InferenceRequest(BaseModel):
previous_character_text: str
previous_speaker: str # ['trump', 'kamala', 'player']
current_speaker: str # ['trump', 'kamala']
class InferenceResponse(BaseModel):
generated_text: str
anger: float
audio: str
class EngagementRequest(BaseModel):
pass
class EngagementResponse(BaseModel):
engagement: int
class CardsVoiceResponse(BaseModel):
presenter_question: str
audio: str
class CardsVoiceRequest(BaseModel):
previous_character_text: str
previous_speaker: str
card_id: int
class CardsResponse(BaseModel):
cards: str
class CardsRequest(BaseModel):
pass
class StartRequest(BaseModel):
"""
Available name for now: ['trump', 'kamala']
"""
candidate_1_name: str
candidate_2_name: str
class StartResponse(BaseModel):
status: str
|