Update app.py

app.py

@@ -114,10 +114,16 @@ class AtlasInferenceApp:
             return "⚠️ Please select and load a model first"
 
         try:
-            # Get the system prompt for the current model
-            system_prompt = st.session_state.current_model["config"]["system_prompt"]
+            # Debugging: Check if config and system_prompt exist
+            if "config" not in st.session_state.current_model:
+                return "⚠️ Model configuration not found. Please load the model again."
+
+            system_prompt = st.session_state.current_model["config"].get("system_prompt", "Default system prompt")
             if not system_prompt:
-                return "⚠️ System prompt not found for the selected model."
+                system_prompt = "Default system prompt"  # Fallback if system_prompt is None
+
+            # Debugging: Print the system prompt for verification
+            st.write(f"System Prompt: {system_prompt}")
 
             # Add the system instruction to guide the model's behavior
             prompt = f"{system_prompt}\n\n### Instruction:\n{message}\n\n### Response:"