Dropdown menu for department added

app.py

@@ -19,7 +19,7 @@ The structure of the script will look like this:
 - Conclusion
 
 You'll only focus on the Hook, Title and Intro. Here are instructions for each:
- 
+
 The hook has only two goals, it must:
 1. Catch the audience's attention and convince them to watch more of the video.
 2. Serve as a setup for the video's content. 
@@ -184,9 +184,10 @@ By leveraging the power of ChatGPT-4, HR professionals can create job advertisem
 
 Remember, in the digital age of recruitment, your job ad is your first impression. So make it count!"""
 
+
 # Function to make API call
 def api_call(messages, temperature=0.9, model="gpt-4-1106-preview"):
-    return openai.ChatCompletion.create( 
+    return openai.ChatCompletion.create(
         messages=messages,
         temperature=temperature,
         model=model
@@ -194,7 +195,7 @@ def api_call(messages, temperature=0.9, model="gpt-4-1106-preview"):
 
 
 # Function to be called by Gradio interface
-def process_inputs(EPA_title, Extra_input):
+def process_inputs(EPA_title, Extra_input, Department):
     # Check if EPA_title is empty
     if not EPA_title:
         return manual
@@ -205,7 +206,8 @@ def process_inputs(EPA_title, Extra_input):
             {"role": "assistant", "content": AssistantPrompt_10},
             {"role": "user", "content": UserPrompt_11},
             {"role": "assistant", "content": AssistantPrompt_11},
-            {"role": "user", "content": f"""Topic of the tutorial: {EPA_title}"""}
+            {"role": "user", "content": f"""Topic of the tutorial: {EPA_title}
+            The tutorial should be aimed at employees at a {Department} department."""}
         ]
         Script_1st_half = api_call(stepOne, 0.7)
 
@@ -213,7 +215,8 @@ def process_inputs(EPA_title, Extra_input):
             {"role": "system", "content": SystemPrompt_20},
             {"role": "user", "content": UserPrompt_20},
             {"role": "assistant", "content": AssistantPrompt_20},
-            {"role": "user", "content": f"""Topic of the tutorial: {EPA_title}"""}
+            {"role": "user", "content": f"""Topic of the tutorial: {EPA_title}
+            The tutorial should be aimed at employees at a {Department} department."""}
         ]
         Script_2nd_half = api_call(stepTwo, 0.7)
         Script_complete = Script_1st_half + "\n\n" + Script_2nd_half
@@ -228,7 +231,8 @@ def process_inputs(EPA_title, Extra_input):
             {"role": "user", "content": UserPrompt_11},
             {"role": "assistant", "content": AssistantPrompt_11},
             {"role": "user", "content": f"""Topic of the tutorial: {EPA_title} 
-            Additional instructions: {Extra_input}"""}
+            Additional instructions: {Extra_input}
+            The tutorial is aimed at employees at a {Department} department."""}
         ]
         Script_1st_half = api_call(stepOne, 0.7)
 
@@ -238,20 +242,43 @@ def process_inputs(EPA_title, Extra_input):
             {"role": "assistant", "content": AssistantPrompt_20},
             {"role": "user", "content": f"""Topic of the tutorial: {EPA_title}
             Additional instructions: {Extra_input}
-            
+            The tutorial is aimed at employees at a {Department} department.
+
             {Script_1st_half}"""}
         ]
         Script_2nd_half = api_call(stepTwo, 0.7)
         Script_complete = Script_1st_half + "\n\n" + Script_2nd_half
         return Script_complete
 
-# Create the Gradio interface 
+departments = {
+    "3D": "3D Modeling",  # Label and value differ
+    "Client Support": "Client Support",
+    "Data Entry & Back Office": "Data Entry & Back Office",
+    "Education Administration (ENI)": "Education Administration",  # Label and value differ
+    "Finance": "Finance",
+    "Marketing": "Marketing",
+    "HR": "HR",
+    "IT": "IT",
+    "Legal": "Legal",
+    "Product Management": "Product Management",
+    "Project Management/Coordination": "Project Management/Coordination",
+    "Process Management": "Process Management",
+    "QA": "QA",
+    "R&D": "R&D",
+    "Sales": "Sales",
+    "Software": "Software",
+    "Video Production": "Video Production",
+    "Visual Management": "Visual Management"
+}
+
+# Create the Gradio interface
 iface = gr.Interface(
-    fn=process_inputs, 
+    fn=process_inputs,
     inputs=[
         gr.Textbox(lines=2, label='EPA title Here... (use the phrasing: "How to do X with ChatGPT")'),
-        gr.Textbox(lines=2, label='Extra input Here... (optional, put any extra requirements or relevant context)')
-        ], 
+        gr.Textbox(lines=2, label='Extra input Here... (optional, put any extra requirements or relevant context)'),
+        gr.Dropdown(choices=departments, label="Select Team", searchable=True)
+    ],
     outputs=gr.Textbox(label="Script", show_copy_button=True)
 )