Update gradio_app.py

gradio_app.py

@@ -26,15 +26,18 @@ Powered by <a href="https://github.com/abetlen/llama-cpp-python">llama-cpp-pytho
 # Loading prompt
 with open('system.prompt', 'r', encoding='utf-8') as f:
     prompt = f.read()
+with open('system.message', 'r', encoding='utf-8') as f:
+    system_message = f.read()
 
 def generate_answer(request: str, max_tokens: int = 256, custom_prompt: str = None):
     logs = f"Request: {request}\nMax tokens: {max_tokens}\nCustom prompt: {custom_prompt}\n"
     try:
         maxTokens = max_tokens if 16 <= max_tokens <= 256 else 64
-        if isinstance(custom_prompt, str) and len(custom_prompt.strip()) > 1 and custom_prompt.strip() not in ['', None, ' ']:
-            userPrompt = custom_prompt.replace("{prompt}", request)
-        else:
-            userPrompt = prompt.replace("{prompt}", request)
+        userPrompt = prompt.replace("{prompt}", request)
+        userPrompt = userPrompt.replace(
+            "{system_message}",
+            custom_prompt if isinstance(custom_prompt, str) and len(custom_prompt.strip()) > 1 and custom_prompt.strip() not in ['', None, ' ']) else system_message
+        )
         logs += f"\nFinal prompt: {userPrompt}\n"
     except:
         return "Not enough data! Check that you passed all needed data.", logs