Rulga commited on
Commit
426daef
·
1 Parent(s): de134bf

Update system prompt handling and improve logging messages

Browse files
Files changed (1) hide show
  1. app.py +17 -20
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, # Используем переданный промпт вместо DEFAULT_SYSTEM_MESSAGE
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"Ошибка сохранения промпта: {str(e)}"
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
- gr.Markdown("")
902
-
903
- with gr.Column(scale=8): # Увеличиваем масштаб для ширины
904
  system_prompt = gr.TextArea(
905
- label="System Prompt (редактирование изменит поведение бота)",
906
- value=saved_system_prompt,
907
- placeholder="Enter system prompt...",
908
- lines=8 # Увеличиваем количество строк
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