Update app.py

app.py

@@ -10,7 +10,7 @@ sessions = {}
 ASSISTANT_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/API.png"  
 USER_PIC_PATH = "https://huggingface.co/spaces/PLRMB/P-MSQ-API-PREVIEW/resolve/main/usr.png"
 
-def authorize(user, api_key):
+def authorize(user, api_key, system_message):
     test_data = {
         "user": user,
         "key": api_key
@@ -36,7 +36,8 @@ def authorize(user, api_key):
                     "headers": {   
                         "authorization": api_key,
                         "Content-Type": 'application/json'
-                    }
+                    },
+                    "system_message": system_message 
                 }
                 
             return True
@@ -54,11 +55,11 @@ def authorize(user, api_key):
         return False
 
 
-def respond(message, api_key, system_message, max_tokens, top_p, temperature):
+def respond(message, api_key, max_tokens, top_p, temperature):
     session = sessions.get(api_key, {})
     history = session.get("history", [])
     headers = session.get("headers", {})
-
+    system_message = session.get("system_message", "You are P-MSQ (Messaging Service Query), a friendly AI Chatbot that can help in any situations.")
     messages = []
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
         if user_message:
@@ -142,8 +143,10 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
         send_btn = gr.Button("Send")
         regen_btn = gr.Button("Clear")
 
-        system_message = gr.Textbox(value="You are P-MSQ (Messaging Service Query), a friendly AI Chatbot that can help in any situations.", label="System message")
-        
+        system_instructions_input = gr.Textbox(placeholder="Enter custom instructions (optional)", 
+                                               label="Custom System Instructions", 
+                                               lines=2)
+        save_instructions_btn = gr.Button("Save Instructions")        
         gr.Markdown("### Settings")
         
         max_tokens = gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens")
@@ -189,8 +192,17 @@ with gr.Blocks(css=".chatbox {height: 400px; overflow-y: auto; border: 1px solid
         else:
             return gr.update(visible=True), gr.update(visible=False), auth_status.update(value="Invalid userid/token")
 
+    def save_custom_instructions(api_key, custom_instructions):
+        if api_key in sessions:
+            sessions[api_key]["system_message"] = custom_instructions  
+            return "Instructions updated!"
+        else:
+            return "Session not found."
 
+            
     auth_button.click(authorize_and_proceed, inputs=[api_user_input, api_key_input], outputs=[auth_view, chat_view, chatbot_output, history_state])
+    save_instructions_btn.click(save_custom_instructions, inputs=[api_key_input, system_instructions_input], outputs=auth_status)
+
 
 if __name__ == "__main__":
     demo.queue = False