Update app.py

app.py

@@ -35,41 +35,58 @@ class RaindropSearchBot:
         logger.info("RaindropSearchBot initialized")
         
     def generate_search_query(self, user_request: str) -> str:
+        """Convert user request to a tailored search query using OpenAI."""
         logger.info(f"Generating search query for request: {user_request}")
-        prompt = f"""
-        Convert the following request into a focused search query for Raindrop.io.
-        Extract key concepts and create a search query that will work well with the Raindrop API.
-        Include relevant operators and consider both exact matches and related terms.
         
-        User Request: {user_request}
+        prompt = f"""
+        Convert this request into a simple search query for finding bookmarks about this topic.
+        Use quotes for exact phrases and OR between alternatives.
+        Only return the search query itself, no explanations.
+        Keep it concise and focused.
         
-        Format the search query for maximum relevance and recall.
+        Request: {user_request}
         """
         
         try:
             response = self.client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=[{"role": "user", "content": prompt}],
-                temperature=0.5,
-                max_tokens=1000
+                temperature=0.3,  # Lower temperature for more focused results
+                max_tokens=100    # Reduced tokens since we only need the query
             )
-            generated_query = response.choices[0].message.content
-            logger.info(f"Generated search query: {generated_query}")
-            return generated_query
+            search_query = response.choices[0].message.content.strip()
+            logger.info(f"Generated search query: {search_query}")
+            return search_query
         except Exception as e:
             logger.error(f"Error generating search query: {str(e)}")
-            raise
+            # Fallback to a simple search if query generation fails
+            return user_request
 
     def search_raindrop(self, search_query: str) -> List[Dict]:
-        logger.info(f"Searching Raindrop with query: {search_query}")
+        """Search Raindrop.io with the generated query."""
+        logger.info(f"Original search query: {search_query}")
+        
+        # Extract the actual search terms from between the backticks
+        try:
+            if "```" in search_query:
+                search_terms = search_query.split("```")[1].strip()
+            else:
+                search_terms = search_query.strip()
+                
+            # Clean up the search terms
+            search_terms = search_terms.replace('\n', ' ').strip()
+            logger.info(f"Cleaned search terms: {search_terms}")
+        except Exception as e:
+            logger.error(f"Error processing search query: {e}")
+            search_terms = search_query
         
         headers = {
             "Authorization": f"Bearer {self.raindrop_api_token}",
             "Content-Type": "application/json"
         }
         
+        # Test the API connection first
         try:
-            # First, let's test the API connection and token
             test_response = requests.get(
                 "https://api.raindrop.io/rest/v1/user",
                 headers=headers
@@ -78,16 +95,15 @@ class RaindropSearchBot:
             if test_response.status_code != 200:
                 logger.error(f"API test failed: {test_response.text}")
                 return []
-
+    
             # Search parameters
             params = {
-                "search": search_query,
+                "search": search_terms,
                 "page": 0,
                 "perpage": 50,
                 "sort": "-created"
             }
             
-            # Log the full request details
             logger.info(f"Making request to Raindrop API with params: {params}")
             
             # Make the search request