Update main.py

main.py

@@ -212,6 +212,7 @@ def get_streams():
     top_p = 0.95
     repetition_penalty = 1.0
 
+
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -220,30 +221,17 @@ def get_streams():
         do_sample=True,
         seed=42,
     )
-
-    # 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", ...]
-    No additional formatting, no extra text, just the JSON array.
+    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
     """
-
     formatted_prompt = format_prompt(prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=False)
-
-    # Debugging: Print raw API response
-    print("Raw API Response:", stream)
-
-    # Clean and validate response
-    try:
-        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})