Upload app.py

app.py

@@ -21,11 +21,16 @@ def compressit(original_text, compressor1, ratio, maxlength):
     else:
         compressor = PromptCompressor(type='SCCompressor', lang='en', model='gpt2', device='cpu')
 
-    if compressor1 != "SCRL":
-        compressed_prompt = compressor.compressgo(original_prompt=original_text, ratio=float(ratio), max_length=int(maxlength))
+    if compressor1 == "Selective Context":
+        compressed_prompt = compressor.compressgo(original_prompt=original_text, ratio=float(ratio))
     else:
-        compressed_prompt = compressor.compressgo(original_prompt=original_text, ratio=float(ratio), max_length=int(maxlength))
-    return compressed_prompt["compressed_prompt"]
+        if maxlength:
+            compressed_prompt = compressor.compressgo(original_prompt=original_text, ratio=float(ratio), max_length=int(maxlength))
+        else:
+            gr.Warning("max_length is needed for this type of compressor. Please fill in and try again.")
+            return "max_length is missing."
+    gr.Warning("The prompt is generating, please wait patiently as it may take a long time to generate during busy hours.")
+    return compressed_prompt["compressed_prompt"], compressed_prompt["ratio"]
 
 
 demo = gr.Interface(
@@ -36,10 +41,11 @@ demo = gr.Interface(
             ["Selective Context", "LLMLingua", "LongLLMLingua", "SCRL", "KiS"], label="compressor", info="Choose your compressor here. \n Currently, we cannot support the online demo for LLMLingua and LongLLMLingua due to the Huggingface Token issue."
         ),
         gr.Textbox(lines=1, placeholder="Enter the compression ratio here...", info="Ratio only works for Selective Context, LLMLingua and LongLLMLingua."),
-        gr.Textbox(lines=1, placeholder="Enter the max_length parameter if you are using SCRL or KiS", label="max_length", info="If you are using SCRL or KiS, fill in the parameter, if not, just ignore this.\n Hint: For SCRL, max_length should be shorter than the lenght of original prompt; For KiS, max_length should be longer than it.")
+        gr.Textbox(lines=1, placeholder="Enter the max_length parameter (integer) if you are using SCRL or KiS", label="max_length", info="If you are using SCRL or KiS, fill in the parameter, if not, just ignore this.\n Hint: For SCRL, max_length should be shorter than the lenght of original prompt; For KiS, max_length should be longer than it.")
     ],
     outputs=[
-        gr.Textbox(lines=1, info="Please wait patiently when proceeding it may take more than 2 minutes to generate since we are using CPUs for free.")
+        gr.Textbox(lines=1, info="Please wait patiently when proceeding it may take more than 2 minutes to generate since we are using CPUs for free."),
+        gr.Textbox(lines=1, info="With the compression ratio of: ")
     ]
 )