Update main.py

main.py

@@ -211,7 +211,6 @@ def get_streams():
     top_p = 0.95
     repetition_penalty = 1.0
 
-
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -220,17 +219,25 @@ def get_streams():
         do_sample=True,
         seed=42,
     )
-    prompt = f""" prompt: 
-    You need to act like as recommendation engine.
-    List all 40+ streams/branches in like computer science, chemical engineering, aerospace , etc
-    Note: Output should be list in below format:
-    [branch1, branch2, branch3,...]
-    Return only answer not prompt and unnecessary stuff, also dont add any special characters or punctuation marks
-    """
+    
+    prompt = """You need to act as a recommendation engine.
+    List all 40+ streams/branches in computer science, chemical engineering, aerospace engineering, etc.
+    Note: The output should be a valid JSON list in the format:
+    ["Artificial Intelligence", "Computer Science", "Data Science", "Cyber Security", ...]
+    Return only the list, no additional text, and ensure it is properly formatted as JSON."""
+    
     formatted_prompt = format_prompt(prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
-    return jsonify({"ans": stream})
+
+    # Ensure the response is correctly formatted as a JSON list
+    try:
+        stream_list = json.loads(stream)  # Convert string to list
+    except json.JSONDecodeError:
+        return jsonify({"error": "Invalid response format"}), 500
+    
+    return jsonify({"ans": stream_list})
+
     
 
 @app.route('/get_education_profiles', methods=['GET'])