Update app.py

app.py

@@ -196,58 +196,83 @@ def get_question(context,answer,model,tokenizer):
 
 import gradio as gr
 import random
+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)
-  random.shuffle(np)
-  print ("\n\nNoun phrases",np)
-
-  output="<b>Select/Match the correct answers to the given questions.</b><br/>"
-  answerlist=[]
-
-  for answer in np:
-    answerlist.append(answer)
-    random.shuffle(answerlist)   
-  print(answerlist)
+    
+  words_text = len(re.findall(r'\w+', context))
+  words_subject = len(re.findall(r'\w+', subject))
 
-  for answer in answerlist:
-    output = output + "<b style='margin-right:30px;'>" + " "*5 +answer.capitalize()+  "</b>"
-  output = output + "<br/>" 
+  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)
+    random.shuffle(np)
+    print ("\n\nNoun phrases",np)
+
+    output="<b>Select/Match the correct answers to the given questions.</b><br/>"
+    answerlist=[]
+
+    for answer in np:
+      answerlist.append(answer)
+      random.shuffle(answerlist)   
+    print(answerlist)
+
+    for answer in answerlist:
+      output = output + "<b style='margin-right:30px;'>" + " "*5 +answer.capitalize()+  "</b>"
+    output = output + "<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> Q "+ str(i) + ")  " + ques + "</b><br/>"
-    i += 1
+    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> Q "+ str(i) + ")  " + ques + "</b><br/>"
+      i += 1
+
+    output = output + "<br/><b>" + "Correct Answer Key:</b><br/>"
+
+    i=1
+    for answer in np:
+      # output= output + ques + "\n" + "Ans: "+answer.capitalize() + "\n\n"
+      output = output + "<b style='color:green;'>Ans "+ str(i) + ": " +answer.capitalize()+  "</b>"
+      # random.shuffle(output)
+      output = output + "<br/>" 
+      i += 1
+    
+    # mycursor = mydb.cursor()
+    # timedate = datetime.datetime.now()
 
-  output = output + "<br/><b>" + "Correct Answer Key:</b><br/>"
+    # sql = "INSERT INTO matchtexts (subject, input, output, timedate) VALUES (%s,%s, %s,%s)"
+    # val = (subject, context, output, timedate)
+    # mycursor.execute(sql, val)
 
-  i=1
-  for answer in np:
-    # output= output + ques + "\n" + "Ans: "+answer.capitalize() + "\n\n"
-    output = output + "<b style='color:green;'>Ans "+ str(i) + ": " +answer.capitalize()+  "</b>"
-    # random.shuffle(output)
-    output = output + "<br/>" 
-    i += 1
+    # mydb.commit()
 
-  
-  return output
+    # 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"])
+  inputs=[context,subject, total], outputs=output,
+  # css=".gradio-container {background-image: url('file=blue.jpg')}",
+  allow_flagging="never",flagging_options=["Save Data"])
   
 # iface.launch(debug=True, share=True)
 
@@ -301,27 +326,50 @@ 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 matchfiles (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 generated
+      return generated
 
-# filecreate(file,2)
 
 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.launch(debug=True, show_api=False)