Spaces:
Running
Running
Update system prompt handling and improve logging messages
Browse files
app.py
CHANGED
|
@@ -579,7 +579,7 @@ def respond_and_clear(message, history, conversation_id, system_prompt):
|
|
| 579 |
message=message,
|
| 580 |
history=history if history else [],
|
| 581 |
conversation_id=conversation_id,
|
| 582 |
-
system_message=system_prompt, #
|
| 583 |
max_tokens=params['max_length'],
|
| 584 |
temperature=params['temperature'],
|
| 585 |
top_p=params['top_p']
|
|
@@ -784,13 +784,13 @@ def save_system_prompt(prompt_text):
|
|
| 784 |
try:
|
| 785 |
preferences = load_user_preferences()
|
| 786 |
|
| 787 |
-
#
|
| 788 |
if "system_prompt" not in preferences:
|
| 789 |
preferences["system_prompt"] = {}
|
| 790 |
|
| 791 |
preferences["system_prompt"]["current"] = prompt_text
|
| 792 |
|
| 793 |
-
#
|
| 794 |
json_content = json.dumps(preferences, indent=2)
|
| 795 |
api = HfApi(token=HF_TOKEN)
|
| 796 |
api.upload_file(
|
|
@@ -800,10 +800,10 @@ def save_system_prompt(prompt_text):
|
|
| 800 |
repo_type="dataset"
|
| 801 |
)
|
| 802 |
|
| 803 |
-
return "
|
| 804 |
except Exception as e:
|
| 805 |
logger.error(f"Error saving system prompt: {str(e)}")
|
| 806 |
-
return f"
|
| 807 |
|
| 808 |
def initialize_app():
|
| 809 |
"""Initialize app with user preferences"""
|
|
@@ -836,6 +836,7 @@ def initialize_app():
|
|
| 836 |
system_prompt_text = preferences["system_prompt"]["current"]
|
| 837 |
|
| 838 |
logger.info(f"App initialized with model: {ACTIVE_MODEL['name']}")
|
|
|
|
| 839 |
return selected_model, system_prompt_text
|
| 840 |
|
| 841 |
def initialize_chat_evaluator():
|
|
@@ -892,25 +893,21 @@ with gr.Blocks() as demo:
|
|
| 892 |
)
|
| 893 |
submit_btn = gr.Button("Send", variant="primary")
|
| 894 |
clear_btn = gr.Button("Clear")
|
| 895 |
-
|
| 896 |
-
# Завершаем предыдущий блок с колонкой
|
| 897 |
-
# Создаем новый блок для системного промпта в центре и шире
|
| 898 |
with gr.Row(equal_height=True):
|
| 899 |
with gr.Column(scale=1):
|
| 900 |
-
|
| 901 |
-
|
| 902 |
-
|
| 903 |
-
with gr.Column(scale=8): # Увеличиваем масштаб для ширины
|
| 904 |
system_prompt = gr.TextArea(
|
| 905 |
-
|
| 906 |
-
|
| 907 |
-
|
| 908 |
-
|
| 909 |
-
|
| 910 |
-
|
| 911 |
with gr.Column(scale=1):
|
| 912 |
-
|
| 913 |
-
gr.Markdown("")
|
| 914 |
|
| 915 |
|
| 916 |
# Add event handlers
|
|
|
|
| 579 |
message=message,
|
| 580 |
history=history if history else [],
|
| 581 |
conversation_id=conversation_id,
|
| 582 |
+
system_message=system_prompt, # Using provided prompt instead of default
|
| 583 |
max_tokens=params['max_length'],
|
| 584 |
temperature=params['temperature'],
|
| 585 |
top_p=params['top_p']
|
|
|
|
| 784 |
try:
|
| 785 |
preferences = load_user_preferences()
|
| 786 |
|
| 787 |
+
# Add prompt to preferences
|
| 788 |
if "system_prompt" not in preferences:
|
| 789 |
preferences["system_prompt"] = {}
|
| 790 |
|
| 791 |
preferences["system_prompt"]["current"] = prompt_text
|
| 792 |
|
| 793 |
+
# Save preferences
|
| 794 |
json_content = json.dumps(preferences, indent=2)
|
| 795 |
api = HfApi(token=HF_TOKEN)
|
| 796 |
api.upload_file(
|
|
|
|
| 800 |
repo_type="dataset"
|
| 801 |
)
|
| 802 |
|
| 803 |
+
return "System prompt saved successfully"
|
| 804 |
except Exception as e:
|
| 805 |
logger.error(f"Error saving system prompt: {str(e)}")
|
| 806 |
+
return f"Error saving prompt: {str(e)}"
|
| 807 |
|
| 808 |
def initialize_app():
|
| 809 |
"""Initialize app with user preferences"""
|
|
|
|
| 836 |
system_prompt_text = preferences["system_prompt"]["current"]
|
| 837 |
|
| 838 |
logger.info(f"App initialized with model: {ACTIVE_MODEL['name']}")
|
| 839 |
+
logger.info(f"Chat histories will be saved to: {DATASET_CHAT_HISTORY_PATH}")
|
| 840 |
return selected_model, system_prompt_text
|
| 841 |
|
| 842 |
def initialize_chat_evaluator():
|
|
|
|
| 893 |
)
|
| 894 |
submit_btn = gr.Button("Send", variant="primary")
|
| 895 |
clear_btn = gr.Button("Clear")
|
| 896 |
+
|
|
|
|
|
|
|
| 897 |
with gr.Row(equal_height=True):
|
| 898 |
with gr.Column(scale=1):
|
| 899 |
+
gr.Markdown("") # Empty column for centering
|
| 900 |
+
|
| 901 |
+
with gr.Column(scale=8):
|
|
|
|
| 902 |
system_prompt = gr.TextArea(
|
| 903 |
+
label="System Prompt (editing will change bot behavior)",
|
| 904 |
+
value=saved_system_prompt,
|
| 905 |
+
placeholder="Enter system prompt...",
|
| 906 |
+
lines=8
|
| 907 |
+
)
|
| 908 |
+
|
| 909 |
with gr.Column(scale=1):
|
| 910 |
+
gr.Markdown("") # Empty column for centering
|
|
|
|
| 911 |
|
| 912 |
|
| 913 |
# Add event handlers
|