From 557782185517e44a614fdebcdbfe4776f5a8771b Mon Sep 17 00:00:00 2001
From: Russell Ballestrini <russell.ballestrini@gmail.com>
Date: Thu, 4 Jul 2024 10:57:30 -0400
Subject: [PATCH] save username when searching

I CAN program offline without an LLM. ; )
	modified:   app.py
	modified:   templates/base.html
	modified:   templates/search.html
---
 app.py                | 56 +++++++++++++++++++++++++------------------
 templates/base.html   |  4 +++-
 templates/search.html | 14 +++++++++++
 3 files changed, 50 insertions(+), 24 deletions(-)

diff --git a/app.py b/app.py
index 5b37dea..5ae36d6 100644
--- a/app.py
+++ b/app.py
@@ -1,8 +1,9 @@
-#import eventlet
-#eventlet.monkey_patch()
+# import eventlet
+# eventlet.monkey_patch()
 
 import gevent
 from gevent import monkey
+
 monkey.patch_all()
 
 
@@ -28,7 +29,7 @@ app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
 
 db = SQLAlchemy(app)
 
-#socketio = SocketIO(app, async_mode="eventlet")
+# socketio = SocketIO(app, async_mode="eventlet")
 socketio = SocketIO(app, async_mode="gevent")
 
 # Global dictionary to keep track of cancellation requests
@@ -141,13 +142,18 @@ def chat(room_name):
 @app.route("/search")
 def search_page():
     keywords = request.args.get("keywords", "")
+    username = request.args.get("username", "guest")
     if not keywords:
-        return render_template("search_results.html", results=[], error="Keywords are required")
+        return render_template(
+            "search.html", results=[], username=username, error="Keywords are required"
+        )
 
     # Call the function to search messages
     search_results = search_messages(keywords)
 
-    return render_template("search.html", results=search_results, error=None)
+    return render_template(
+        "search.html", results=search_results, username=username, error=None
+    )
 
 
 def search_messages(keywords):
@@ -165,7 +171,9 @@ def search_messages(keywords):
         room = Room.query.get(message.room_id)
         if room:
             # Calculate the score based on the number of occurrences of all keywords
-            score = sum(message.content.lower().count(keyword) for keyword in keyword_list)
+            score = sum(
+                message.content.lower().count(keyword) for keyword in keyword_list
+            )
 
             # Extract snippets with context around each occurrence of the keywords
             snippets = []
@@ -180,7 +188,9 @@ def search_messages(keywords):
 
                     # Calculate the snippet range
                     snippet_start = max(0, start_index - 25)
-                    snippet_end = min(len(message.content), start_index + len(keyword) + 25)
+                    snippet_end = min(
+                        len(message.content), start_index + len(keyword) + 25
+                    )
                     snippet = message.content[snippet_start:snippet_end]
 
                     snippets.append(snippet)
@@ -196,7 +206,7 @@ def search_messages(keywords):
                     "room_title": room.title,
                     "snippets": [],
                     "username": message.username,
-                    "score": 0
+                    "score": 0,
                 }
 
             search_results[room.id]["snippets"].append(snippet_text)
@@ -533,32 +543,32 @@ def group_consecutive_roles(messages):
         return []
 
     grouped_messages = []
-    current_role = messages[0]['role']
+    current_role = messages[0]["role"]
     current_content = []
 
     for message in messages:
-        if message['role'] == current_role:
-            current_content.append(message['content'])
+        if message["role"] == current_role:
+            current_content.append(message["content"])
         else:
-            grouped_messages.append({
-                'role': current_role,
-                'content': ' '.join(current_content)
-            })
-            current_role = message['role']
-            current_content = [message['content']]
+            grouped_messages.append(
+                {"role": current_role, "content": " ".join(current_content)}
+            )
+            current_role = message["role"]
+            current_content = [message["content"]]
 
     # Append the last grouped message
-    grouped_messages.append({
-        'role': current_role,
-        'content': ' '.join(current_content)
-    })
+    grouped_messages.append(
+        {"role": current_role, "content": " ".join(current_content)}
+    )
 
     return grouped_messages
 
 
 def chat_claude(
-    #username, room_name, model_name="anthropic.claude-3-5-sonnet-20240620-v1:0"
-    username, room_name, model_name="anthropic.claude-3-sonnet-20240229-v1:0"
+    # username, room_name, model_name="anthropic.claude-3-5-sonnet-20240620-v1:0"
+    username,
+    room_name,
+    model_name="anthropic.claude-3-sonnet-20240229-v1:0",
 ):
     with app.app_context():
         room = get_room(room_name)
diff --git a/templates/base.html b/templates/base.html
index bc8b63e..3d9657a 100644
--- a/templates/base.html
+++ b/templates/base.html
@@ -172,6 +172,7 @@
   <div id="search-form" style="width: 90%;">
       <form action="/search" method="get">
           <input type="text" id="search-keywords" name="keywords" placeholder="Search for keywords..." style="width: 90%;">
+          <input type="hidden" id="username" name="username" value="{{ username }}">
       </form>
   </div>
   <!-- Chatroom list -->
@@ -196,13 +197,14 @@
       // Function to perform the search
       function performSearch() {
           const keywords = document.getElementById("search-keywords").value;
+          const username = document.getElementById("username").value;
           if (!keywords) {
               alert("Please enter keywords to search.");
               return;
           }
 
           // Navigate to the search results page with the keywords as a query parameter
-          window.location.href = `/search?keywords=${encodeURIComponent(keywords)}`;
+          window.location.href = `/search?keywords=${encodeURIComponent(keywords)}&username=${username}`;
       }
 
       // Add event listener for keyword search the "Enter" key
diff --git a/templates/search.html b/templates/search.html
index c4d3183..d58c7ac 100644
--- a/templates/search.html
+++ b/templates/search.html
@@ -3,6 +3,19 @@
 {% block title %}Search Results{% endblock %}
 
 {% block content %}
+<style>
+    /* Add styles for the search results */
+    #search-results {
+        margin-top: 20px;
+    }
+    .search-result {
+        border: 1px solid #e1e1e1;
+        border-radius: 5px;
+        padding: 10px;
+        margin-bottom: 10px;
+    }
+</style>
+
 <div id="search-results">
     {% if error %}
         <div class="search-result">{{ error }}</div>
@@ -22,3 +35,4 @@
     {% endif %}
 </div>
 {% endblock %}
+