gabrielchua commited on
Commit
da7d0fa
1 Parent(s): cecf748

update gradio UI

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -32,7 +32,7 @@ class Dialogue(BaseModel):
32
  """The dialogue between the host and guest."""
33
 
34
  scratchpad: str
35
- participants: List[str]
36
  dialogue: List[DialogueItem]
37
 
38
 
@@ -49,13 +49,16 @@ def generate_podcast(file: str) -> Tuple[str, str]:
49
 
50
  # Process the dialogue
51
  audio_segments = []
52
- transcript = ""
53
  total_characters = 0
54
 
55
  for line in llm_output.dialogue:
56
  logger.info(f"Generating audio for {line.speaker}: {line.text}")
57
- transcript_line = f"{line.speaker}: {line.text}"
58
- transcript += transcript_line + "\n\n"
 
 
 
59
  total_characters += len(line.text)
60
 
61
  # Get audio file path
@@ -99,10 +102,11 @@ demo = gr.Interface(
99
  ],
100
  outputs=[
101
  gr.Audio(label="Audio", format="mp3"),
102
- gr.Textbox(label="Transcript"),
103
  ],
104
  allow_flagging="never",
105
  api_name=False,
 
106
  )
107
 
108
  if __name__ == "__main__":
 
32
  """The dialogue between the host and guest."""
33
 
34
  scratchpad: str
35
+ name_of_guest: str
36
  dialogue: List[DialogueItem]
37
 
38
 
 
49
 
50
  # Process the dialogue
51
  audio_segments = []
52
+ transcript = "" # start with an empty transcript
53
  total_characters = 0
54
 
55
  for line in llm_output.dialogue:
56
  logger.info(f"Generating audio for {line.speaker}: {line.text}")
57
+ if line.speaker == "Host (Jane)":
58
+ speaker = f"**Jane**: {line.text}"
59
+ else:
60
+ speaker = f"**{llm_output.name_of_guest}**: {line.text}"
61
+ transcript += speaker + "\n\n"
62
  total_characters += len(line.text)
63
 
64
  # Get audio file path
 
102
  ],
103
  outputs=[
104
  gr.Audio(label="Audio", format="mp3"),
105
+ gr.Markdown(label="Transcript"),
106
  ],
107
  allow_flagging="never",
108
  api_name=False,
109
+ theme=gr.themes.Soft()
110
  )
111
 
112
  if __name__ == "__main__":