Update app.py

app.py

@@ -176,45 +176,69 @@ def get_question(context,answer,model,tokenizer):
 
 
 import gradio as gr
+import re
 
-context = gr.Textbox(lines=10, placeholder="Enter paragraph/content here...", label="Text")
+context = gr.Textbox(lines=10, placeholder="Enter paragraph/content here...", label="Enter your content (words input must be more than 150 words).")
 total = gr.Slider(1,10, value=1,step=1, label="Total Number Of Questions")
-subject = gr.Textbox(placeholder="Enter subject/title here...", label="Text")
+subject = gr.Textbox(placeholder="Enter subject/title here...", label="Enter your title (title must contain 1 word)")
 
 
 output = gr.Markdown(  label="Question and Answers")
 
 
 def generate_question_text(context,subject,total):
-  summary_text = summarizer(context,summary_model,summary_tokenizer)
-  for wrp in wrap(summary_text, 150):
-    print (wrp)
-  np =  get_keywords(context,summary_text,total)
-  print ("\n\nNoun phrases",np)
-  output="<b style='color:black;'>Answer the following true/false questions. Select the correct answer.</b><br><br>"
-
-  i=1
-  for answer in np:
-    ques = get_question(summary_text,answer,question_model,question_tokenizer)
-    # output= output + ques + "\n" + "Ans: "+answer.capitalize() + "\n\n"
-    output = output + "<b style='color:black;'>Q"+ str(i) + ") " + ques + "</b> <br>"
-    # output = output + "<br>"
-    output = output + "<b>" + "a) True <br></b>"
-    output = output + "<b>" + "b) False </b>"
-
-    output = output + "<br>"
-    i += 1
-
-  return output
+  words_text = len(re.findall(r'\w+', context))
+  words_subject = len(re.findall(r'\w+', subject))
+
+  if (words_text < 150):
+    raise gr.Error("Invalid Input (Words limit must be more than 150 words).")
+        # print("Number of words:", words)
+    
+  elif (words_subject < 1):
+      raise gr.Error("Invalid Input (Title must be one or more than one word).")
+    
+  else:
+    summary_text = summarizer(context,summary_model,summary_tokenizer)
+    for wrp in wrap(summary_text, 150):
+      print (wrp)
+    np =  get_keywords(context,summary_text,total)
+    print ("\n\nNoun phrases",np)
+    output="<b style='color:black;'>Answer the following true/false questions. Select the correct answer.</b><br><br>"
+
+    i=1
+    for answer in np:
+      ques = get_question(summary_text,answer,question_model,question_tokenizer)
+      # output= output + ques + "\n" + "Ans: "+answer.capitalize() + "\n\n"
+      output = output + "<b style='color:black;'>Q"+ str(i) + ") " + ques + "</b> <br>"
+      # output = output + "<br>"
+      output = output + "<b>" + "a) True <br></b>"
+      output = output + "<b>" + "b) False </b>"
+
+      output = output + "<br>"
+      i += 1
+    
+    # mycursor = mydb.cursor()
+    # timedate = datetime.datetime.now()
+
+    # sql = "INSERT INTO truetexts (subject, input, output, timedate) VALUES (%s,%s, %s,%s)"
+    # val = (subject, context, output, timedate)
+    # mycursor.execute(sql, val)
+
+    # mydb.commit()
+    # print(mycursor.rowcount, "record inserted.")
+
+    return output
 
 iface = gr.Interface(
   fn=generate_question_text, 
   inputs=[context,subject, total], 
   outputs=output,
-  allow_flagging="manual",flagging_options=["Save Data"])
+  # css=".gradio-container {background-image: url('file=blue.jpg')}",
+  allow_flagging="never",flagging_options=["Save Data"])
 
 # iface.launch(debug=True, share=True)
 
+
 def generate_question(context,subject,total):
   summary_text = summarizer(context,summary_model,summary_tokenizer)
   for wrp in wrap(summary_text, 150):
@@ -248,26 +272,51 @@ def filecreate(x,subject,total):
     with open(x.name) as fo:  
       text = fo.read()
     #   print(text)
-    generated = generate_question(text,subject, total)
 
+    words_text = len(re.findall(r'\w+', text))
+    words_subject = len(re.findall(r'\w+', subject))
+
+
+    if (words_text < 150):
+        raise gr.Error("Invalid Input (Words limit must be more than 150 words).")
+        # print("Number of words:", words)
+    
+    elif (words_subject < 1):
+        raise gr.Error("Invalid Input (Title must be one or more than one word).")
+    
+    else:
+      generated = generate_question(text,subject, total)
+
+      # mycursor = mydb.cursor()
+
+      # timedate= datetime.datetime.now()
+
+      # sql = "INSERT INTO truefiles (subject, input, output, timedate) VALUES (%s,%s, %s,%s)"
+      # val = (subject, text, generated, timedate)
+      # mycursor.execute(sql, val)
+
+      # mydb.commit()
+
+      # print(mycursor.rowcount, "record inserted.")
 
-    # return text
-    return generated
+      # return text
+      return generated
 
 import gradio as gr
 
 context = gr.HTML(label="Text")
-file = gr.File()
+file = gr.File(label="Upload your file (File must contain more than 150 words).")
 total = gr.Slider(1,10, value=1,step=1, label="Total Number Of Questions")
-subject = gr.Textbox(placeholder="Enter subject/title here...", label="Text")
+subject = gr.Textbox(placeholder="Enter subject/title here...", label="Enter your title (title must contain 1 word).")
 
 fface = gr.Interface(  
   fn=filecreate, 
   inputs=[file,subject,total], 
   outputs=context, 
-  allow_flagging="manual",flagging_options=["Save Data"])
+  # css=".gradio-container {background-image: url('file=blue.jpg')}",
+  allow_flagging="never",flagging_options=["Save Data"])
 
 # fface.launch(debug=True, share=True)
 
-demo = gr.TabbedInterface([iface, fface], ["Text", "Upload File"],)
-demo.launch(debug=True)
+demo = gr.TabbedInterface([iface, fface], ["Text", "Upload File"])
+demo.launch(debug=True, show_api=False)