Update main.py

main.py

@@ -212,7 +212,6 @@ def get_streams():
     top_p = 0.95
     repetition_penalty = 1.0
 
-
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -221,17 +220,30 @@ 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 like Computer Science, Chemical Engineering, Aerospace, etc.
+    Note: Output should be a **valid JSON list**:
+    ["branch1", "branch2", "branch3", ...]
+    Return only the list, **do not return text or explanations**.
     """
+    
     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})
+
+    try:
+        # Ensure output is valid JSON
+        cleaned_data = stream.strip()
+        if not cleaned_data.startswith("["):
+            raise ValueError("Invalid response format")
+
+        parsed_data = json.loads(cleaned_data)
+        return jsonify({"ans": parsed_data})  # Send as proper JSON
+    except Exception as e:
+        return jsonify({"error": "Invalid response format", "details": str(e)})
+