Update app.py

app.py

@@ -23,7 +23,8 @@ def process_email_wrapper(
     max_tokens,
     temperature,
     top_p,
-    custom_guidelines
+    custom_guidelines,
+    identity
 ):
     """
     Wrapper function to call the email processor with all necessary parameters.
@@ -40,7 +41,8 @@ def process_email_wrapper(
         top_p,
         PASSWORD,
         CLIENTS,
-        custom_guidelines
+        custom_guidelines,
+        identity
     )
 
 def check_password(input_password):
@@ -153,6 +155,14 @@ with gr.Blocks(title="Email Assistant") as demo:
                             minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"
                         )
                 
+                with gr.Row():
+                    identity = gr.Textbox(
+                        value="",
+                        label="Reply Identity (optional)",
+                        placeholder="Enter the perspective this email is being written from (e.g., 'Sales Manager', 'Customer Support')",
+                        lines=1
+                    )
+                
                 with gr.Row():
                     with gr.Column():
                         email_guidelines = gr.Textbox(
@@ -228,7 +238,8 @@ with gr.Blocks(title="Email Assistant") as demo:
                 max_tokens,
                 temperature,
                 top_p,
-                email_guidelines
+                email_guidelines,
+                identity
             ],
             outputs=[assistant_response, suggestions]
         )