Update src/services/processor.py

src/services/processor.py

@@ -215,4 +215,63 @@ def select_technologies(problem_technology_list):
 
     if best_set is None:
         return set()
-    return set(best_set)
+    return set(best_set)
+
+
+def search_prior_art(technologies_input: list, data: str, type: str) -> json:
+    """
+    Searches for prior art patents online that solve a given technical problem
+    using a set of specified technologies, leveraging the Gemini model's search capabilities.
+    """
+
+    technology_titles = [tech['title'] for tech in technologies_input]
+
+    if type == "problem":
+        prompt = f"Find prior art patents or research paper online that address the technical problem: '{data}'. " \
+             
+    elif type == "constraints":
+        prompt = f"Find prior art patents or research paper online that address those constraints: '{data}'. " \
+
+    prompt += f"Using any combination of the following technologies:  {', '.join(technology_titles)}. " \
+         f"Specifically look for patents that integrate multiple of these technologies. " \
+         f"Indicate for each document found what technologies is used inside of it from the provided list"
+
+    client,config = set_gemini()
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=prompt,
+        config=config,
+    )
+
+    return response
+
+def add_citations_and_collect_uris(response):
+    try:
+        print(response)
+        text = response.text
+        supports = response.candidates[0].grounding_metadata.grounding_supports
+        chunks = response.candidates[0].grounding_metadata.grounding_chunks
+
+        sorted_supports = sorted(supports, key=lambda s: s.segment.end_index, reverse=True)
+
+        uris_added = set()
+
+        for support in sorted_supports:
+            end_index = support.segment.end_index
+            if support.grounding_chunk_indices:
+                citation_links = []
+                for i in support.grounding_chunk_indices:
+                    if i < len(chunks):
+                        uri = chunks[i].web.uri
+                        # Add URI only if not already in text or collected
+                        if uri not in text and uri not in uris_added:
+                            citation_links.append(f"[{i + 1}]({uri})")
+                            uris_added.add(uri)
+                if citation_links:
+                    citation_string = ", ".join(citation_links)
+                    text = text[:end_index] + citation_string + text[end_index:]
+    except Exception as e:
+        print(f"Error : {e}")
+
+    return {"content": text,"uris": list(uris_added)}