Update app.py

app.py

@@ -1,23 +1,30 @@
-import gradio as gr
+
+import warnings
 from huggingface_hub import InferenceClient
 
+warnings.filterwarnings('ignore')
+
 # Initialize the language model
 generator = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def generate_script(host_name="John", listener_location="City", causes_climate_change="human activities",
-                    co2_level=400, effects_climate_change="rising temperatures", sea_level_rise=0.1,
-                    warming_rate=0.2, potential_solutions="renewable energy", individual_role="reduce carbon footprint",
-                    call_to_action="act now"):
+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
-    prompt_template = f"""INTRODUCTION: {host_name}, good morning! This is {listener_location}'s local radio station. Today we're talking about an issue that affects us all - climate change. It's a pressing issue that requires our immediate attention...
-    CAUSES OF CLIMATE CHANGE: The causes of climate change are {causes_climate_change}. Today, the level of CO2 in our atmosphere is {co2_level}, which is concerning...
-    EFFECTS OF CLIMATE CHANGE: 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...
-    POTENTIAL SOLUTIONS: But don't worry, there are solutions. {potential_solutions} are all steps we can take to mitigate these effects...
-    INDIVIDUAL ROLE: Each one of us plays a role in combatting climate change. Even small actions can make a big difference. In fact, our location, {listener_location}, is particularly vulnerable to climate change due to its geographical features...
-    CALL TO ACTION: 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: In conclusion, climate change 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!"""
-
-    # Generate the script
+    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!"
+
+    # 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']
 
     # Split the script into sections
@@ -33,15 +40,19 @@ def generate_script(host_name="John", listener_location="City", causes_climate_c
 
     return script
 
-# Create a Gradio interface with default values
+# 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")
+                        
+
 iface = gr.Interface(fn=generate_script,
                    inputs=[gr.Textbox(label="Host Name", value="John"),
                            gr.Textbox(label="Listener Location", value="City"),
                            gr.Textbox(label="Causes Climate Change", value="human activities"),
-                           gr.Number(label="CO2 Level", default=400),
+                           gr.Number(label="CO2 Level", value=400),
                            gr.Textbox(label="Effects Climate Change", value="rising temperatures"),
-                           gr.Number(label="Sea Level Rise", default=0.1),
-                           gr.Number(label="Warming Rate", default=0.2),
+                           gr.Number(label="Sea Level Rise", value=0.1),
+                           gr.Number(label="Warming Rate", value=0.2),
                            gr.Textbox(label="Potential Solutions", value="renewable energy"),
                            gr.Textbox(label="Individual Role", value="reduce carbon footprint"),
                            gr.Textbox(label="Call To Action", value="act now")],