Update app.py

app.py

@@ -78,8 +78,6 @@ def sentence_embedder(sentences, model_path):
   sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask']) #outputs a (1, 384) tensor representation of input text
   return sentence_embeddings
     
-
-    
 def add_text(history, text):
     history = history + [(text, None)]
     return history, ""
@@ -238,8 +236,10 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
     
     with gr.Tab("Claim Drafter"):
         gr.Markdown(""" 
-        Use this tool to expand your idea into the technical language of a patent claim.        
+        Use this tool to expand your idea into the technical language of a patent claim. You can specify the type of claim you want using the dropdown menu.
         """)
+        gr.Dropdown(["Apparatus Claim", "Method of Use Claim", "Method Claim", ], label='Choose Claim Type Here')        
+        
         with gr.Row(scale=1, min_width=600):
             text1 = gr.Textbox(label="Input",
                               placeholder='Type in your idea here!')
@@ -247,10 +247,8 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
    
     with gr.Tab("Description Generator"):
         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.
- 
-        """)
+         """)
         gr.Dropdown(["Generate Description", "Generate Abstract", "Benefits of the invention"], label='Choose Generation Type Here')
         with gr.Row(scale=1, min_width=600):
                     
@@ -258,24 +256,23 @@ with gr.Blocks(title='Claimed', theme=theme) as demo:
                               placeholder='Type in your idea here!')
             text2 = gr.Textbox(label="Output")
 
-    with gr.Tab("Knowledge Graph"):
-        gr.Markdown(""" 
-        Use this tool to generate a knowledge graph of your invention. This will help highlight the links between features. 
-
-        """)
-        with gr.Row(scale=1, min_width=600):
-            text1 = gr.Textbox(label="Input",
-                              placeholder='Type in your idea here!')
-            text2 = gr.Textbox(label="Output")
+    # with gr.Tab("Knowledge Graph"):
+    #     gr.Markdown(""" 
+    #     Use this tool to generate a knowledge graph of your invention. This will help highlight the links between features. 
+    #     """)
+    #     with gr.Row(scale=1, min_width=600):
+    #         text1 = gr.Textbox(label="Input",
+    #                           placeholder='Type in your idea here!')
+    #         text2 = gr.Textbox(label="Output")
 
-    with gr.Tab("Prosecution Ideator"):
-        gr.Markdown(""" 
-        Use this tool to generate ideas for how to overcome objections to novelty and inventive step. Outputs are in the problem-solution format. 
-        """)
-        with gr.Row(scale=1, min_width=600):
-            text1 = gr.Textbox(label="Input",
-                              placeholder='Type in your idea here!')
-            text2 = gr.Textbox(label="Output")
+    # with gr.Tab("Prosecution Ideator"):
+    #     gr.Markdown(""" 
+    #     Use this tool to generate ideas for how to overcome objections to novelty and inventive step. Outputs are in the problem-solution format. 
+    #     """)
+    #     with gr.Row(scale=1, min_width=600):
+    #         text1 = gr.Textbox(label="Input",
+    #                           placeholder='Type in your idea here!')
+    #         text2 = gr.Textbox(label="Output")
 
     # with gr.Tab("Claimed Infill"):
     #     gr.Markdown("""