Update app.py

app.py

@@ -19,20 +19,20 @@ def get_course_recommendations(stanine, gwa, strand, hobbies):
     
     # Validate inputs
     if not stanine or not gwa or not strand or not hobbies:
-        return "Please fill in all fields", "", ""
+        return "Please fill in all fields", "", "", "", "", ""
     
     try:
         stanine = int(stanine)
         gwa = float(gwa)
         
         if stanine < 1 or stanine > 9:
-            return "Stanine must be between 1-9", "", ""
+            return "Stanine must be between 1-9", "", "", "", "", ""
         
         if gwa < 75 or gwa > 100:
-            return "GWA must be between 75-100", "", ""
+            return "GWA must be between 75-100", "", "", "", "", ""
         
         if strand not in ["STEM", "ABM", "HUMSS", "GAS", "TVL"]:
-            return "Strand must be one of: STEM, ABM, HUMSS, GAS, TVL", "", ""
+            return "Strand must be one of: STEM, ABM, HUMSS, GAS, TVL", "", "", "", "", ""
         
         # Store current user input
         current_user_input = {
@@ -46,51 +46,100 @@ def get_course_recommendations(stanine, gwa, strand, hobbies):
         recommendations = recommender.predict_course(stanine, gwa, strand, hobbies)
         current_recommendations = recommendations
         
-        # Format recommendations
-        result_text = "🎓 Course Recommendations:\n\n"
-        for i, (course, confidence) in enumerate(recommendations, 1):
-            confidence_percent = confidence * 100
-            result_text += f"{i}. {course} (Confidence: {confidence_percent:.1f}%)\n"
-        
-        # Show top recommendation
-        top_course = recommendations[0][0] if recommendations else "No recommendations"
-        confidence = recommendations[0][1] * 100 if recommendations else 0
+        # Format top 3 recommendations
+        if len(recommendations) >= 3:
+            course1 = f"{recommendations[0][0]} (Confidence: {recommendations[0][1]*100:.1f}%)"
+            course2 = f"{recommendations[1][0]} (Confidence: {recommendations[1][1]*100:.1f}%)"
+            course3 = f"{recommendations[2][0]} (Confidence: {recommendations[2][1]*100:.1f}%)"
+        elif len(recommendations) == 2:
+            course1 = f"{recommendations[0][0]} (Confidence: {recommendations[0][1]*100:.1f}%)"
+            course2 = f"{recommendations[1][0]} (Confidence: {recommendations[1][1]*100:.1f}%)"
+            course3 = "No third recommendation available"
+        elif len(recommendations) == 1:
+            course1 = f"{recommendations[0][0]} (Confidence: {recommendations[0][1]*100:.1f}%)"
+            course2 = "No second recommendation available"
+            course3 = "No third recommendation available"
+        else:
+            course1 = "No recommendations available"
+            course2 = ""
+            course3 = ""
         
-        return result_text, top_course, f"{confidence:.1f}%"
+        return course1, course2, course3, None, None, None
         
     except ValueError as e:
-        return f"Invalid input: {str(e)}", "", ""
+        return f"Invalid input: {str(e)}", "", "", None, None, None
     except Exception as e:
-        return f"Error getting recommendations: {str(e)}", "", ""
+        return f"Error getting recommendations: {str(e)}", "", "", None, None, None
 
-def rate_recommendation(rating):
-    """Rate the current recommendation (like/dislike)"""
+def submit_all_ratings(course1_rating, course2_rating, course3_rating):
+    """Submit ratings for all three recommendations"""
     global current_recommendations, current_user_input
     
     if not current_recommendations or not current_user_input:
         return "No recommendations to rate. Please get recommendations first."
     
     try:
-        # Convert rating to numeric
-        rating_value = 1 if rating == "👍 Like" else 0
-        
-        # Get the top recommendation
-        top_course = current_recommendations[0][0]
-        
-        # Add feedback to database
-        success = recommender.add_feedback(
-            course=top_course,
-            stanine=current_user_input['stanine'],
-            gwa=current_user_input['gwa'],
-            strand=current_user_input['strand'],
-            rating=rating_value,
-            hobbies=current_user_input['hobbies']
-        )
+        results = []
+        ratings_submitted = 0
+        
+        # Rate first recommendation
+        if course1_rating and len(current_recommendations) >= 1:
+            rating_value = 1 if course1_rating == "👍 Like" else 0
+            course = current_recommendations[0][0]
+            success = recommender.add_feedback(
+                course=course,
+                stanine=current_user_input['stanine'],
+                gwa=current_user_input['gwa'],
+                strand=current_user_input['strand'],
+                rating=rating_value,
+                hobbies=current_user_input['hobbies']
+            )
+            if success:
+                results.append(f"✅ Rating for '{course}' recorded")
+                ratings_submitted += 1
+            else:
+                results.append(f"❌ Failed to record rating for '{course}'")
         
-        if success:
-            return f"✅ Thank you for your feedback! Your rating for '{top_course}' has been recorded."
+        # Rate second recommendation
+        if course2_rating and len(current_recommendations) >= 2:
+            rating_value = 1 if course2_rating == "👍 Like" else 0
+            course = current_recommendations[1][0]
+            success = recommender.add_feedback(
+                course=course,
+                stanine=current_user_input['stanine'],
+                gwa=current_user_input['gwa'],
+                strand=current_user_input['strand'],
+                rating=rating_value,
+                hobbies=current_user_input['hobbies']
+            )
+            if success:
+                results.append(f"✅ Rating for '{course}' recorded")
+                ratings_submitted += 1
+            else:
+                results.append(f"❌ Failed to record rating for '{course}'")
+        
+        # Rate third recommendation
+        if course3_rating and len(current_recommendations) >= 3:
+            rating_value = 1 if course3_rating == "👍 Like" else 0
+            course = current_recommendations[2][0]
+            success = recommender.add_feedback(
+                course=course,
+                stanine=current_user_input['stanine'],
+                gwa=current_user_input['gwa'],
+                strand=current_user_input['strand'],
+                rating=rating_value,
+                hobbies=current_user_input['hobbies']
+            )
+            if success:
+                results.append(f"✅ Rating for '{course}' recorded")
+                ratings_submitted += 1
+            else:
+                results.append(f"❌ Failed to record rating for '{course}'")
+        
+        if ratings_submitted > 0:
+            return f"Thank you! {ratings_submitted} rating(s) submitted successfully.\n\n" + "\n".join(results)
         else:
-            return "❌ Failed to record feedback. Please try again."
+            return "Please select at least one rating before submitting."
             
     except Exception as e:
         return f"Error recording feedback: {str(e)}"
@@ -116,15 +165,15 @@ def create_interface():
             with gr.Column(scale=1):
                 gr.Markdown("### 📝 Your Profile")
                 
-                stanine_input = gr.Slider(
-                    minimum=1, maximum=9, step=1, value=5,
+                stanine_input = gr.Textbox(
                     label="Stanine Score (1-9)",
+                    placeholder="Enter your stanine score (1-9)",
                     info="Your stanine score from standardized tests"
                 )
                 
-                gwa_input = gr.Slider(
-                    minimum=75, maximum=100, step=0.1, value=85,
+                gwa_input = gr.Textbox(
                     label="GWA (75-100)",
+                    placeholder="Enter your GWA (75-100)",
                     info="Your Grade Weighted Average"
                 )
                 
@@ -146,33 +195,43 @@ def create_interface():
                 train_model_btn = gr.Button("🤖 Train Model", variant="secondary")
             
             with gr.Column(scale=1):
-                gr.Markdown("### 🎓 Recommendations")
+                gr.Markdown("### 🎓 Top 3 Course Recommendations")
                 
-                recommendations_output = gr.Textbox(
-                    label="Course Recommendations",
-                    lines=8,
+                # Display top 3 recommendations
+                course1_output = gr.Textbox(
+                    label="1st Recommendation",
                     interactive=False
                 )
                 
-                top_course_output = gr.Textbox(
-                    label="Top Recommendation",
-                    interactive=False
+                course1_rating = gr.Radio(
+                    choices=["👍 Like", "👎 Dislike"],
+                    label="Rate 1st Recommendation",
+                    interactive=True
                 )
                 
-                confidence_output = gr.Textbox(
-                    label="Confidence Score",
+                course2_output = gr.Textbox(
+                    label="2nd Recommendation", 
                     interactive=False
                 )
                 
-                gr.Markdown("### 👍 Rate Your Recommendation")
+                course2_rating = gr.Radio(
+                    choices=["👍 Like", "👎 Dislike"],
+                    label="Rate 2nd Recommendation",
+                    interactive=True
+                )
+                
+                course3_output = gr.Textbox(
+                    label="3rd Recommendation",
+                    interactive=False
+                )
                 
-                rating_buttons = gr.Radio(
+                course3_rating = gr.Radio(
                     choices=["👍 Like", "👎 Dislike"],
-                    label="How do you rate this recommendation?",
+                    label="Rate 3rd Recommendation",
                     interactive=True
                 )
                 
-                rate_btn = gr.Button("Submit Rating", variant="secondary")
+                submit_ratings_btn = gr.Button("Submit All Ratings", variant="primary")
                 
                 rating_feedback = gr.Textbox(
                     label="Rating Feedback",
@@ -183,12 +242,12 @@ def create_interface():
         get_recommendations_btn.click(
             fn=get_course_recommendations,
             inputs=[stanine_input, gwa_input, strand_input, hobbies_input],
-            outputs=[recommendations_output, top_course_output, confidence_output]
+            outputs=[course1_output, course2_output, course3_output, course1_rating, course2_rating, course3_rating]
         )
         
-        rate_btn.click(
-            fn=rate_recommendation,
-            inputs=[rating_buttons],
+        submit_ratings_btn.click(
+            fn=submit_all_ratings,
+            inputs=[course1_rating, course2_rating, course3_rating],
             outputs=[rating_feedback]
         )