Update app.py

app.py

@@ -672,9 +672,11 @@ def generate_audio(text, description="Thomas speaks with emphasis and excitement
     prompt = tokenizer(preprocess(text), return_tensors="pt").to(device)
 
     set_seed(SEED)
-    input_features = model.get_input_features(prompt.input_ids)  # Ensure we have input_features
+    input_ids = inputs.input_ids.to(device)
+    prompt_input_ids = prompt.input_ids.to(device)
 
-    generation = model.generate(input_features=input_features, input_ids=inputs.input_ids)
+    # Use the generate method to get the audio features
+    generation = model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
     audio_arr = generation.cpu().numpy().squeeze()
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
@@ -684,12 +686,6 @@ def generate_audio(text, description="Thomas speaks with emphasis and excitement
     logging.debug(f"Audio saved to {temp_audio_path}")
     return temp_audio_path
 
-def install_parler_tts():
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/huggingface/parler-tts.git"])
-
-# Call the function to install parler-tts
-install_parler_tts()
-
 # Check if the token is already set in the environment variables
 hf_token = os.getenv("HF_TOKEN")
 
@@ -739,7 +735,7 @@ def fetch_local_events():
     url = f'https://serpapi.com/search.json?engine=google_events&q=Events+in+Birmingham&hl=en&gl=us&api_key={api_key}'
 
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200){
         events_results = response.json().get("events_results", [])
         events_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Local Events</h2>
@@ -760,7 +756,7 @@ def fetch_local_events():
             }
         </style>
         """
-        for index, event in enumerate(events_results):
+        for index, event in enumerate(events_results){
             title = event.get("title", "No title")
             date = event.get("date", "No date")
             location = event.get("address", "No location")
@@ -772,11 +768,13 @@ def fetch_local_events():
             </div>
             """
         return events_html
-    else:
+    }
+    else{
         return "<p>Failed to fetch local events</p>"
+    }
 
 def fetch_local_weather():
-    try:
+    try{
         api_key = os.environ['WEATHER_API']
         url = f'https://weather.visualcrossing.com/VisualCrossingWebServices/rest/services/timeline/birmingham?unitGroup=metric&include=events%2Calerts%2Chours%2Cdays%2Ccurrent&key={api_key}'
         response = requests.get(url)
@@ -786,10 +784,12 @@ def fetch_local_weather():
         current_conditions = jsonData.get("currentConditions", {})
         temp_celsius = current_conditions.get("temp", "N/A")
         
-        if temp_celsius != "N/A":
+        if temp_celsius != "N/A"){
             temp_fahrenheit = int((temp_celsius * 9/5) + 32)
-        else:
+        }
+        else{
             temp_fahrenheit = "N/A"
+        }
             
         condition = current_conditions.get("conditions", "N/A")
         humidity = current_conditions.get("humidity", "N/A")
@@ -840,8 +840,10 @@ def fetch_local_weather():
         </style>
         """
         return weather_html
-    except requests.exceptions.RequestException as e:
+    }
+    catch (requests.exceptions.RequestException as e){
         return f"<p>Failed to fetch local weather: {e}</p>"
+    }
 
 def get_weather_icon(condition):
     condition_map = {
@@ -990,12 +992,13 @@ def generate_map(location_names):
     
     for location_name in all_addresses:
         geocode_result = gmaps.geocode(location_name)
-        if geocode_result:
+        if geocode_result){
             location = geocode_result[0]['geometry']['location']
             folium.Marker(
                 [location['lat'], location['lng']],
                 tooltip=f"{geocode_result[0]['formatted_address']}"
             ).add_to(m)
+    }
     
     map_html = m._repr_html_()
     return map_html
@@ -1004,7 +1007,7 @@ def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200){
         results = response.json().get("news_results", [])
         news_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Birmingham Today</h2>
@@ -1060,7 +1063,7 @@ def fetch_local_news():
         </script>
         <div id="news-preview" class="news-preview"></div>
         """
-        for index, result in enumerate(results[:7]):
+        for index, result in enumerate(results[:7]){
             title = result.get("title", "No title")
             link = result.get("link", "#")
             snippet = result.get("snippet", "")
@@ -1071,8 +1074,10 @@ def fetch_local_news():
             </div>
             """
         return news_html
-    else:
+    }
+    else{
         return "<p>Failed to fetch local news</p>"
+    }
 
 # Voice Control
 import numpy as np
@@ -1097,18 +1102,22 @@ base_audio_drive = "/data/audio"
 import numpy as np
 
 def transcribe_function(stream, new_chunk):
-    try:
+    try{
         sr, y = new_chunk[0], new_chunk[1]
-    except TypeError:
+    }
+    catch (TypeError){
         print(f"Error chunk structure: {type(new_chunk)}, content: {new_chunk}")
         return stream, "", None
+    }
 
     y = y.astype(np.float32) / np.max(np.abs(y))
 
-    if stream is not None:
+    if stream is not None{
         stream = np.concatenate([stream, y])
-    else:
+    }
+    else{
         stream = y
+    }
 
     result = pipe_asr({"array": stream, "sampling_rate": sr}, return_timestamps=False)
 
@@ -1151,16 +1160,18 @@ def generate_audio_elevenlabs(text):
         }
     }
     response = requests.post(tts_url, headers=headers, json=data, stream=True)
-    if response.ok:
+    if response.ok{
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as f:
-            for chunk in response.iter_content(chunk_size=1024):
+            for chunk in response.iter_content(chunk_size=1024){
                 f.write(chunk)
             temp_audio_path = f.name
         logging.debug(f"Audio saved to {temp_audio_path}")
         return temp_audio_path
-    else:
+    }
+    else{
         logging.error(f"Error generating audio: {response.text}")
         return None
+    }
 
 # Stable Diffusion setup
 pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
@@ -1188,8 +1199,8 @@ def update_images():
 
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     
-    with gr.Row():
-        with gr.Column():
+    with gr.Row(){
+        with gr.Column(){
             state = gr.State()
             
             chatbot = gr.Chatbot([], elem_id="RADAR:Channel 94.1", bubble_full_width=False)
@@ -1212,13 +1223,14 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             # gr.Markdown("<h1 style='color: red;'>Map</h1>", elem_id="location-markdown")
             # location_output = gr.HTML()
             # bot_msg.then(show_map_if_details, [chatbot, choice], [location_output, location_output])
+        }
         
         # with gr.Column():
         #     weather_output = gr.HTML(value=fetch_local_weather())
         #     news_output = gr.HTML(value=fetch_local_news())
         #     news_output = gr.HTML(value=fetch_local_events())
             
-        with gr.Column():
+        with gr.Column(){
             
             image_output_1 = gr.Image(value=generate_image(hardcoded_prompt_1), width=400, height=400)
             image_output_2 = gr.Image(value=generate_image(hardcoded_prompt_2), width=400, height=400)
@@ -1227,6 +1239,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
 
             refresh_button = gr.Button("Refresh Images")
             refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
+        }
      
 demo.queue()
 demo.launch(share=True)