Update app.py

app.py

@@ -182,14 +182,12 @@ def clean_data(input, type='Dataframe'):
         row = ' '.join(wonum)
         return row
 
-
-
-def classifier(userin):
+def classifier(userin, SearchType):
     clean_in = clean_data(userin, type='String')
     in_emb = sentence_embedder(clean_in, 'Model_bert')
     
     Number = 10
-    broad_scope_predictions = broad_scope_class_predictor(class_embeddings, in_emb, Number, Sensitivity='High')
+    broad_scope_predictions = broad_scope_class_predictor(class_embeddings, in_emb, SearchType, Number, Sensitivity='High')
 
     return broad_scope_predictions
     
@@ -292,15 +290,18 @@ def claim_selector(userin, dropd):
 
 def desc_selector(userin, dropd):
     
-    PROMPT = f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.
-    ### Instruction:
-    {dropd} for {userin}
-    ### Response:"""
+    if dropd == "What are the Benefits/Technical Effects":
+        PROMPT =  f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.
+        ### Instruction:
+        What are the Benefits/Technical Effects of the following invention: {userin}
+        ### Response:"""
+    else:
+        PROMPT = f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.
+        ### Instruction:
+        {dropd} for a patent application for the following invention: {userin}
+        ### Response:"""
 
     return PROMPT
-    
-  
-        
 
 ############# GRADIO APP ###############
 
@@ -343,7 +344,7 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
         gr.Markdown(""" 
         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.
          """)
-        Descriptionchoices = gr.Dropdown(["Generate Description", "Generate Abstract", "Benefits of the invention"], label='Choose Generation Type Here')
+        Descriptionchoices = gr.Dropdown(["Generate a Description", "Generate a Abstract", "What are the Benefits/Technical Effects"], label='Choose Generation Type Here')
         with gr.Row(scale=1, min_width=600):
                     
             text1 = gr.Textbox(label="Input",