Update app.py

app.py

@@ -90,28 +90,53 @@ print ("\n")
 """# **UI by using Gradio**"""
 
 import gradio as gr
+import re
 
-context = gr.Textbox(lines=10, placeholder="Enter paragraph/content here...", label="Text")
-subject = gr.Textbox(placeholder="Enter subject/title 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).")
+subject = gr.Textbox(placeholder="Enter subject/title here...", label="Enter your title (title must contain 1 word)")
 output = gr.Markdown(  label="Notes")
 
 def generate_question_text(context,subject):
-  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;'>Notes and key points of the topic are:</b><br>"
-  summary = summary_text
-  output = output+ summary
   
-  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;'>Notes and key points of the topic are:</b><br>"
+    summary = summary_text
+    output = output+ summary
+
+    # mycursor = mydb.cursor()
+    # timedate = datetime.datetime.now()
+
+    # sql = "INSERT INTO notestexts (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], 
-    outputs=[output],
-  allow_flagging="manual",flagging_options=["Save Data"])
+  inputs=[context,subject], outputs=[output], 
+  # css=".gradio-container {background-image: url('file=blue.jpg')}",
+  allow_flagging="never",flagging_options=["Save Data"])
   
 # iface.launch(debug=True, share=True)
 
@@ -139,27 +164,52 @@ def filecreate(x,subjectfile):
     with open(x.name) as fo:  
       text = fo.read()
     #   print(text)
-    generated = generate_question(text,subject)
 
-    return generated
+    words_text = len(re.findall(r'\w+', text))
+    words_subject = len(re.findall(r'\w+', subjectfile))
+
+
+    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)
+
+      # mycursor = mydb.cursor()
+
+      # timedate= datetime.datetime.now()
+
+      # sql = "INSERT INTO notesfiles (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
 
 # filecreate(file)
 
 import gradio as gr
 
 context = gr.HTML(label="Text")
-subjectfile = gr.Textbox(placeholder="Enter subject/title here...", label="Text")
+subjectfile = gr.Textbox(placeholder="Enter subject/title here...", label="Enter your title (title must contain 1 word).")
 
-file = gr.File()
+file = gr.File(label="Upload your file (File must contain more than 150 words).")
 
 fface = gr.Interface(  
   fn=filecreate, 
   inputs=[file,subjectfile], 
   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)