Update app.py

app.py

@@ -1,10 +1,6 @@
-
 import warnings
 from huggingface_hub import InferenceClient
 import gradio as gr
-from huggingface_hub import InferenceClient
-
-
 
 warnings.filterwarnings('ignore')
 
@@ -14,41 +10,35 @@ generator = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 def generate_script(host_name, listener_location, causes_climate_change, co2_level, effects_climate_change,
                     sea_level_rise, warming_rate, potential_solutions, individual_role, call_to_action,
                     TOPIC, DESCRIPTION):
-    # Variables
-    introduction_template = f"{host_name}, good morning! This is {listener_location}'s local radio station. Today we're talking about an issue that affects us all - {TOPIC}. It's a pressing issue that requires our immediate attention..."
-    causes_template = f"The causes of {TOPIC} are {causes_climate_change}. Today, the level of CO2 in our atmosphere is {co2_level}, which is concerning..."
-    effects_template = f"These activities result in {effects_climate_change}, leading to drastic changes in our environment. For instance, sea levels are rising at a rate of {sea_level_rise} per year, and global temperatures are increasing at a rate of {warming_rate} per decade..."
-    solutions_template = f"But don't worry, there are solutions. {potential_solutions} are all steps we can take to mitigate these effects..."
-    role_template = f"Each one of us plays a role in combating {TOPIC}. Even small actions can make a big difference. In fact, our location, {listener_location}, is particularly vulnerable to {TOPIC} due to its geographical features..."
-    action_template = f"So, {listener_location}, why wait? Start taking steps today towards a greener future. Support local businesses that prioritize sustainability, reduce your carbon footprint, and voice your opinion to policy makers..."
-    summary_template = f"In conclusion, {TOPIC} is a serious issue that requires our immediate attention. But by understanding its causes, effects, and potential solutions, we can all play a part in mitigating its impact. Thank you for joining us today, and remember, every small action counts!"
+    try:
+        # Variables and template definitions...
 
-    # Combine templates based on the DESCRIPTION
-    prompt_template = f"""{introduction_template} {causes_template} {effects_template} {solutions_template} {role_template} {action_template} {summary_template}
-    
-    TOPIC: {TOPIC}. DESCRIPTION: {DESCRIPTION}"""
+        # Combine templates based on the DESCRIPTION
+        prompt_template = f"""{introduction_template} {causes_template} {effects_template} {solutions_template} {role_template} {action_template} {summary_template}
+        
+        TOPIC: {TOPIC}. DESCRIPTION: {DESCRIPTION}"""
 
-    # Generate the script using the language model
-    script = generator(prompt_template, max_length=1000)[0]['generated_text']
+        # Generate the script using the language model
+        script = generator(prompt_template, max_length=1000)[0]['generated_text']
 
-    # Split the script into sections
-    sections = script.split("\n")
+        # Split the script into sections
+        sections = script.split("\n")
 
-    # Calculate the word count for each section
-    word_counts = [len(section.split()) for section in sections]
+        # Calculate the word count for each section
+        word_counts = [len(section.split()) for section in sections]
 
-    # Check if any section exceeds the target word count
-    for i, count in enumerate(word_counts):
-        if count > 200:
-            print(f"Warning: Section {i + 1} exceeds the target word count. You may need to shorten this section.")
+        # Check if any section exceeds the target word count
+        for i, count in enumerate(word_counts):
+            if count > 200:
+                print(f"Warning: Section {i + 1} exceeds the target word count. You may need to shorten this section.")
 
-    return script
+        return script
 
-# Example usage
-#generate_script("Host Name", "Listener Location", "Causes of Climate Change", 415, "Effects of Climate Change",
-#                2.5, 1.5, "Potential Solutions", "Individual Role", "Call to Action", "Environmental Issues", "Description of the environmental issues")
-                        
+    except Exception as e:
+        print(f"Error: {e}")
+        return "An error occurred. Please check the console for details."
 
+# Gradio interface setup...
 iface = gr.Interface(fn=generate_script,
                    inputs=[gr.Textbox(label="Host Name", value="John"),
                            gr.Textbox(label="Listener Location", value="City"),
@@ -64,4 +54,3 @@ iface = gr.Interface(fn=generate_script,
 
 # Launch the interface
 iface.launch(debug=True)
-