updated the system prompt

app.py

@@ -31,8 +31,12 @@ client = Groq(api_key=GROQ_API_KEY)
 SYSTEM_PROMPT_TEMPLATE = (
     "You are an advanced AI agent leveraging the Reflexion framework to iteratively improve ideas and responses through up to {reflection_cycles} cycles of reflection. "
     "Your goal is to provide the most meaningful, relevant, and impactful results while autonomously managing the process. Follow the structured workflow below:\n\n"
-    "Instructions:\n\n"
-    "Output the entire response in the following JSON structure:\n"
+    "CRITICAL: Your entire response MUST be a valid JSON object. Follow these strict formatting rules:\n"
+    "1. Use double quotes (\") for all JSON keys and string values\n"
+    "2. Escape any double quotes within string values with a backslash (\\\")\n"
+    "3. Avoid using any unescaped special characters or line breaks within string values\n"
+    "4. Ensure all JSON keys and values are properly enclosed and comma-separated\n\n"
+    "Required JSON Structure:\n"
     "{{\n"
     "  \"initial_response\": \"<Provide the initial response here as a string>\",\n"
     "  \"reflection_cycles\": [\n"
@@ -59,7 +63,8 @@ SYSTEM_PROMPT_TEMPLATE = (
     "Use the feedback from this reflection to refine the response and document it in the JSON structure.\n\n"
     "Final Output:\n"
     "Provide a final, polished response as the \"final_output\" field in the JSON. The response should be thoughtful, comprehensive, and fully address the user's query.\n\n"
-    "Previous Context:\n{history_context}\n"
+    "Previous Context:\n{history_context}\n\n"
+    "REMINDER: Verify that your response is valid JSON before completing. Do not include any text outside of the JSON structure."
 )
 
 # Initialize Streamlit app
@@ -78,9 +83,12 @@ def sanitize_json(json_str):
 # Function to generate responses using the Groq API
 def generate_response(user_input, refined_history):
     try:
+        # Filter out None values and ensure all items are strings
+        valid_history = [str(item) for item in refined_history if item is not None]
+        
         # Limit the number of historical responses
         MAX_HISTORY = 5
-        history_context = " ".join(refined_history[-MAX_HISTORY:])
+        history_context = " ".join(valid_history[-MAX_HISTORY:])
 
         # Format the system prompt with history and current input
         formatted_prompt = SYSTEM_PROMPT_TEMPLATE.format(
@@ -158,14 +166,14 @@ if user_input:
         response = generate_response(user_input, st.session_state.refined_history)
         if response:
             try:
-                refined_response = response["final_output"]
-                
-                # Add the refined response to history
-                st.session_state.refined_history.append(refined_response)
-                
-                # Display the refined response
-                st.chat_message("assistant").markdown(refined_response)
-                st.session_state.messages.append({"role": "assistant", "content": refined_response})
+                refined_response = response.get("final_output", "")
+                if refined_response:  # Only append non-empty responses
+                    # Add the refined response to history
+                    st.session_state.refined_history.append(refined_response)
+                    
+                    # Display the refined response
+                    st.chat_message("assistant").markdown(refined_response)
+                    st.session_state.messages.append({"role": "assistant", "content": refined_response})
             except Exception as e:
                 logging.error(f"Error parsing response: {e}")
                 st.error("Failed to process the response.")