Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ user_input = st.text_area("Enter Patent Data", "Paste your patent details here..
|
|
13 |
# Sidebar for user input
|
14 |
with st.sidebar:
|
15 |
number_of_generated_search_terms = st.number_input("Number of Search Terms to Generate", value=5, min_value=1)
|
16 |
-
|
17 |
model_name = st.text_input("OpenAI Model Name", "gpt-4-0125-preview")
|
18 |
|
19 |
# Fetch API keys from environment variables
|
@@ -41,23 +41,23 @@ else:
|
|
41 |
|
42 |
# Search patents using generated terms
|
43 |
with st.spinner('Searching for patents...'):
|
44 |
-
|
45 |
st.success("Patents search completed!")
|
46 |
st.subheader("Top Patents Found for Search Terms")
|
47 |
|
48 |
-
for search_term, patents in
|
49 |
-
st.write(f"Top {
|
50 |
if len(patents) == 0:
|
51 |
st.write(f"No patents found for search term: {search_term}")
|
52 |
else:
|
53 |
patent_titles = [patent['patentTitle'] for patent in patents]
|
54 |
-
st.json(patent_titles[:
|
55 |
|
56 |
# Check similarity of patents
|
57 |
patents_list = []
|
58 |
with st.spinner('Checking similarities of patents...'):
|
59 |
-
for search_term, patents in
|
60 |
-
patents_list.extend(patents[:
|
61 |
similarities = check_similarity_of_patents(user_input, patents_list)
|
62 |
st.success("Similarity check completed!")
|
63 |
|
|
|
13 |
# Sidebar for user input
|
14 |
with st.sidebar:
|
15 |
number_of_generated_search_terms = st.number_input("Number of Search Terms to Generate", value=5, min_value=1)
|
16 |
+
top_k_patents = st.number_input("Top K Patents to Search", value=10, min_value=1)
|
17 |
model_name = st.text_input("OpenAI Model Name", "gpt-4-0125-preview")
|
18 |
|
19 |
# Fetch API keys from environment variables
|
|
|
41 |
|
42 |
# Search patents using generated terms
|
43 |
with st.spinner('Searching for patents...'):
|
44 |
+
search_terms_patents = search_on_google_patents(generated_search_terms)
|
45 |
st.success("Patents search completed!")
|
46 |
st.subheader("Top Patents Found for Search Terms")
|
47 |
|
48 |
+
for search_term, patents in search_terms_patents.items():
|
49 |
+
st.write(f"Top {top_k_patents} patents found for search term: {search_term}")
|
50 |
if len(patents) == 0:
|
51 |
st.write(f"No patents found for search term: {search_term}")
|
52 |
else:
|
53 |
patent_titles = [patent['patentTitle'] for patent in patents]
|
54 |
+
st.json(patent_titles[:top_k_patents])
|
55 |
|
56 |
# Check similarity of patents
|
57 |
patents_list = []
|
58 |
with st.spinner('Checking similarities of patents...'):
|
59 |
+
for search_term, patents in search_terms_patents.items():
|
60 |
+
patents_list.extend(patents[:top_k_patents])
|
61 |
similarities = check_similarity_of_patents(user_input, patents_list)
|
62 |
st.success("Similarity check completed!")
|
63 |
|