Upload app.py

app.py

@@ -132,7 +132,7 @@ def load_and_prepare_data():
     book_titles = dataset["Book-Title"]
 
 
-def recommend_books(target_book: str, num_recommendations: int = 10) -> str:
+def recommend_books_with_theme(target_book: str, num_recommendations: int = 10, theme= None):
     global dataset, faiss_index, normalized_data, book_titles
 
     if dataset is None or faiss_index is None or normalized_data is None or book_titles is None:
@@ -145,26 +145,30 @@ def recommend_books(target_book: str, num_recommendations: int = 10) -> str:
 
     correlations = compute_correlations_faiss(faiss_index, list(dataset["Book-Title"]), closest_match, num_recommendations)
 
-    recommendations = dataset[dataset["Book-Title"] != target_book].head(num_recommendations)
+    recommendations = dataset[dataset["Book-Title"][correlations['book']] != target_book].head(num_recommendations)
 
     result = f"Top {num_recommendations} recommendations for '{target_book}':\n\n"
     for i, (_, row) in enumerate(recommendations.iterrows(), 1):
         result += f"{i}. {row['book']} (Correlation: {row['corr']:.2f})\n"
+    # Set theme based on user selection
+    theme_mode = "light" if theme == "Light" else "dark"
+    return result, theme_mode
 
-    return result
 
-
-# Create Gradio interface
+# Gradio interface
 iface = gr.Interface(
-    fn=recommend_books,
+    fn=recommend_books_with_theme,
     inputs=[
         gr.Textbox(label="Enter a book title"),
-        gr.Slider(minimum=1, maximum=20, step=1, label="Number of recommendations", value=10)
+        gr.Slider(minimum=1, maximum=20, step=1, label="Number of recommendations", value=10),
+        gr.Dropdown(["Light", "Dark"], label="Theme", value="Light")  # Theme toggle
+    ],
+    outputs=[
+        gr.Textbox(label="Recommendations"),
+        gr.Text(label="Current Theme"),  # Show selected theme
     ],
-    outputs=gr.Textbox(label="Recommendations"),
-    title="Book Recommender",
-    description="Enter a book title to get recommendations based on user ratings and book similarities."
+    title="Book Recommender with Theme Toggle",
+    description="Enter a book title to get recommendations and select a theme (Light/Dark)."
 )
 
-# Launch the app
 iface.launch()