Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from langchain.prompts import PromptTemplate
-from langchain_community.llms import HuggingFacePipeline  # Updated import path
+from langchain_huggingface import HuggingFacePipeline
 from transformers import pipeline
 from bs4 import BeautifulSoup
 import requests
@@ -21,11 +21,12 @@ Summarize the following article content in a clear, warm, and motivational tone
 Summary:
 """)
 
-# Updated chaining method
 summary_chain = summary_prompt | llm
 
 # TTS model setup
 tts_model = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=False, gpu=False)
+SPEAKER_LIST = tts_model.speakers
+DEFAULT_SPEAKER = SPEAKER_LIST[0] if SPEAKER_LIST else None
 
 def extract_main_content(url):
     try:
@@ -39,13 +40,13 @@ def extract_main_content(url):
     except Exception as e:
         return f"Error extracting article content: {str(e)}"
 
-def generate_human_like_audio(text):
+def generate_human_like_audio(text, speaker):
     try:
         temp_dir = tempfile.mkdtemp()
         wav_path = os.path.join(temp_dir, "summary.wav")
         mp3_path = os.path.join(temp_dir, "summary.mp3")
 
-        tts_model.tts_to_file(text=text, file_path=wav_path)
+        tts_model.tts_to_file(text=text, speaker=speaker, file_path=wav_path)
 
         os.system(f"ffmpeg -y -i {wav_path} -codec:a libmp3lame -qscale:a 4 {mp3_path}")
 
@@ -57,7 +58,7 @@ def generate_human_like_audio(text):
         print(f"TTS ERROR: {e}")
         return None, None
 
-def url_to_audio_summary(url):
+def url_to_audio_summary(url, speaker):
     try:
         article_text = extract_main_content(url)
         if article_text.startswith("Error"):
@@ -69,13 +70,13 @@ def url_to_audio_summary(url):
         summary = summary_chain.invoke({"text": article_text})
         summary = summary["text"] if isinstance(summary, dict) and "text" in summary else summary
 
-        wav_path, mp3_path = generate_human_like_audio(summary)
+        wav_path, mp3_path = generate_human_like_audio(summary, speaker)
         return summary, wav_path, mp3_path
     except Exception as e:
         return f"Error: {str(e)}", None, None
 
-def interface_wrapper(url):
-    summary, wav_path, mp3_path = url_to_audio_summary(url)
+def interface_wrapper(url, speaker):
+    summary, wav_path, mp3_path = url_to_audio_summary(url, speaker)
     download_html = ""
     if mp3_path and os.path.exists(mp3_path):
         download_html = f'<a href="file/{os.path.basename(mp3_path)}" download target="_blank">Click to download MP3</a>'
@@ -83,7 +84,10 @@ def interface_wrapper(url):
 
 iface = gr.Interface(
     fn=interface_wrapper,
-    inputs=gr.Textbox(label="Article URL", placeholder="Paste a news/blog URL here..."),
+    inputs=[
+        gr.Textbox(label="Article URL", placeholder="Paste a news/blog URL here..."),
+        gr.Dropdown(choices=SPEAKER_LIST, value=DEFAULT_SPEAKER, label="Select Speaker")
+    ],
     outputs=[
         gr.Textbox(label="Summary"),
         gr.Audio(label="Preacher-style Audio Summary", type="filepath"),