Update app.py

app.py

@@ -36,14 +36,14 @@ def bot(history):
 class_embeddings = pd.read_csv('Embeddings/MainClassEmbeddings.csv')
 
 
-def classifier(userin):
+def classifier(userin, searchlink):
     clean_in = classification.clean_data(userin, type='String')
     in_emb = classification.sentence_embedder(clean_in, 'Model_bert')
 
     Number = 10
     broad_scope_predictions = classification.broad_scope_class_predictor(class_embeddings, in_emb, Number, Sensitivity='High')
 
-    return broad_scope_predictions[1]
+    return broad_scope_predictions[1], searchlink
     
 
 # def generateresponse(history):
@@ -131,7 +131,7 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
         Use this tool to expand your patent claim into a description. You can also use this tool to generate abstracts and give you ideas about the benefit of an invention by changing the settings in the dropdown menu.
  
         """)
-        gr.Dropdown(["Generate Description", "Generate Abstract", "Benefits of the invention"], label='Choose Generation Type Here')
+        gr.Dropdown(["Generate Description", "Generate Abstract", "Benefits of the invention"], label='Choose Generation Type Here', value='Generate Description')
         with gr.Row(scale=1, min_width=600):
                     
             text1 = gr.Textbox(label="Input",
@@ -177,14 +177,14 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
         Click on the link to initiate either an Espacenet or Google Patents classification search using the generated classifications. You can specify which you would like using the dropdown menu. 
         """)
 
-        gr.Dropdown(["Google Patent Search", "Espacenet Patent Search"], label='Choose Search Link Here')        
+        SearchType = gr.Dropdown(["Google Patent Search", "Espacenet Patent Search"], label='Choose Search Type Here', value='Google Patent Search')        
         with gr.Row(scale=1, min_width=600):
             userin = gr.Textbox(label="Input",
                               placeholder='Type in your Claim/Description/Abstract Here')
             output = gr.Textbox(label="Output")
         with gr.Row():
             classify_btn = gr.Button("Classify")
-            classify_btn.click(fn=classifier, inputs=userin, outputs=output)
+            classify_btn.click(fn=classifier, inputs=[userin, SearchType] , outputs=output)
 
 
     gr.Markdown("""