Merge pull request #32 from REMLA24-Team-5/saga-make-new-old-version

new old version
REMLA24-Team-5 · Jun 26, 2024 · a9a9752 · a9a9752
2 parents 3747840 + a24d25e
commit a9a9752
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app-frontend/templates/index.html b/app-frontend/templates/index.html
@@ -40,8 +40,8 @@ <h4 class="modal-title">Do you think this is phishing?</h4>
               <p>Indicate below if you think the URL is phishing. You can then compare your understanding of phishing URLs with the prediction of the model.</p>
             </div>
             <div class="modal-footer">
-              <button name="yesButton" id="yesButton" class="btn btn-danger" type="submit">I think it's phishing</button>
-              <button name="noButton" id="noButton" class="btn btn-success" type="submit">It is innocent</button>
+              <button name="yesButton" id="yesButton" class="btn btn-success" type="submit">I think it's phishing</button>
+              <button name="noButton" id="noButton" class="btn btn-danger" type="submit">It is innocent</button>
             </div>
           </div>
 

diff --git a/app-service/app.py b/app-service/app.py
@@ -49,7 +49,7 @@ def index():
 
                 url_info = connector_to_model_service.get_url_data(url)
 
-                if url_info["prediction"] == "phishing":
+                if url_info["prediction"] == "Phishing":
                     PHISHING_COUNT.inc()
                     if user_thinks_phishing:
                         # User guessed the same as model
@@ -58,7 +58,7 @@ def index():
                     else:
                         # User did not
                         USER_GUESS_AGREE_GAUGE.set(0)
-                elif url_info["prediction"] == "legitimate":
+                elif url_info["prediction"] == "Legitimate":
                     if not user_thinks_phishing:
                         # User guessed the same as model
                         USER_SAME_GUESS_COUNT.inc()