Update app.py

app.py

@@ -14,27 +14,30 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 #     return prompt
 def format_prompt(topic, description, difficulty):
     prompt = (
-        f"You are an expert educator. Generate highly engaging and structured educational content on the topic '{topic}'. "
+        f"You are an expert educator. Generate highly engaging, educational, and structured content on the topic '{topic}'. "
         f"Use the following description as context: '{description}'. "
-        f"The content should be suitable for a '{difficulty}' difficulty level and MUST strictly follow this format:\n\n"
-        f"Title: [The title of the topic]\n\n"
-        f"Section 1:\n"
-        f"Subheading: [A clear subheading summarizing the section]\n"
-        f"Reading Content:\n"
-        f"[Detailed explanation of the section content]\n\n"
-        f"Section 2:\n"
-        f"Subheading: [A clear subheading summarizing the section]\n"
-        f"Reading Content:\n"
-        f"[Detailed explanation of the section content]\n\n"
-        f"Repeat this structure for all additional sections.\n\n"
-        f"Do not generate any hashtags, extra symbols, or additional formats outside the specified structure. "
-        f"Ensure all content is well-structured, concise, and informative."
+        f"The content should be suitable for a '{difficulty}' difficulty level and MUST strictly follow this JSON structure:\n\n"
+        f"{{\n"
+        f"  \"title\": \"[The title of the content]\",\n"
+        f"  \"sections\": [\n"
+        f"    {{\n"
+        f"      \"subheading\": \"[A clear subheading summarizing the section]\",\n"
+        f"      \"content\": \"[Detailed explanation of the section content in plain text]\"\n"
+        f"    }}\n"
+        f"  ]\n"
+        f"}}\n\n"
+        f"Ensure that:\n"
+        f"1. The output is a valid JSON object.\n"
+        f"2. The JSON structure includes a 'title' and a list of 'sections', where each section contains 'subheading' and 'content'.\n"
+        f"3. Do not include any extra text, explanations, or symbols outside the JSON structure.\n"
+        f"4. Validate the JSON before returning it to ensure it follows standard JSON formatting."
     )
     return prompt
 
 
 
 
+
 # The function to generate learning content based on the inputs
 def generate_learning_content(topic, description, difficulty, temperature=0.9, max_new_tokens=2000, top_p=0.95, repetition_penalty=1.2):
     temperature = max(float(temperature), 1e-2)  # Ensure minimum temperature