Update main.py

main.py

@@ -220,24 +220,31 @@ def get_streams():
         do_sample=True,
         seed=42,
     )
-    
-    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:
+
+    # Updated prompt to enforce strict JSON output
+    prompt = """You are a recommendation engine.
+    List 40+ academic streams like "Artificial Intelligence", "Chemical Engineering", etc.
+    Output **must be** a valid JSON array **without extra text**:
     ["Artificial Intelligence", "Computer Science", "Data Science", "Cyber Security", ...]
-    Return only the list, no additional text, and ensure it is properly formatted as JSON."""
-    
+    No additional formatting, no extra text, just the JSON array.
+    """
+
     formatted_prompt = format_prompt(prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
 
-    # Ensure the response is correctly formatted as a JSON list
+    # Debugging: Print raw API response
+    print("Raw API Response:", stream)
+
+    # Clean and validate response
     try:
-        stream_list = json.loads(stream)  # Convert string to list
-    except json.JSONDecodeError:
+        stream_cleaned = stream.strip().replace("\n", "").replace("`", "").replace("json", "")
+        stream_list = json.loads(stream_cleaned)  # Convert string to Python list
+        return jsonify({"ans": stream_list})
+    except json.JSONDecodeError as e:
+        print("JSON Decode Error:", e)
         return jsonify({"error": "Invalid response format"}), 500
-    
-    return jsonify({"ans": stream_list})
+