Refactor control function and update interface

app.py

@@ -22,21 +22,23 @@ def summarize_text(text: str, bullet_points: int, conclusion: bool) -> str:
     summary = api.text_generation(prompt, max_new_tokens=250, do_sample=True)
     return summary["generated_text"]
 
-def control(audio_file: gr.AudioFile, text: str, bullet_points: int, conclusion: bool) -> str:
+def control(audio_file, text: str, bullet_points: int, conclusion: bool) -> str:
     if audio_file:
         text = transcript_audio(audio_file)
     summary = summarize_text(text, bullet_points, conclusion)
     return summary
 # make a simeple interface, where the user can input a text and get a summary or input an audio file and get a transcript and a summary
 iface = gr.Interface(
-    fn=summarize_text,
+    fn=control,
     inputs=[
         gr.components.Audio(label="Audio file"),
         gr.components.Textbox(lines=5, label="Text"),
-        gr.components.Slider(minimum=1, maximum=10, value=5, step=1, label="Number of bullet points"),
+        gr.components.Slider(
+            minimum=1, maximum=10, value=5, step=1, label="Number of bullet points"
+        ),
         gr.components.Checkbox(label="Add conclusion"),
     ],
-    outputs=gr.components.Textbox(label="Summary")
+    outputs=gr.components.Textbox(label="Summary"),
 )
 
 iface.launch()