Refactor app.py to enable SemanticSearcher and UpvotePredictor functionality

app.py

@@ -5,9 +5,8 @@ import pandas as pd
 from datasets import load_dataset
 
 from src.genai import GenAI
-
-# from src.semantic_searcher import SemanticSearcher
-# from src.upvote_predictor import UpvotePredictor
+from src.semantic_searcher import SemanticSearcher
+from src.upvote_predictor import UpvotePredictor
 
 # Load the dataset
 dataset_counsel_chat = load_dataset("nbertagnolli/counsel-chat")
@@ -53,19 +52,19 @@ examples = "\n".join(
 
 # Initialize the SemanticSearcher
 genai = GenAI()
-# upvote_predictor = UpvotePredictor("models/bert_model")
-# _ = SemanticSearcher(df_counsel_chat_topic)
+upvote_predictor = UpvotePredictor("models/bert_model")
+_ = SemanticSearcher(df_counsel_chat_topic)
 
 
 def get_output(question: str, question_context: str = None) -> str:
     answer, topic = genai.generate_content(
         question, question_context, unique_topics, examples
     )
-    return (answer, topic, "Yes", pd.DataFrame())
-    # upvote_prediction = upvote_predictor.get_upvote_prediction(
-    #     question, answer, question_context
-    # )
-    # return (answer, topic, upvote_prediction[0], upvote_prediction[1])
+    # return (answer, topic, "Yes", pd.DataFrame())
+    upvote_prediction = upvote_predictor.get_upvote_prediction(
+        question, answer, question_context
+    )
+    return (answer, topic, upvote_prediction[0], upvote_prediction[1])
 
 
 demo = gr.Interface(