Changes to UI

app.py

@@ -135,7 +135,7 @@ def generate_answer(question):
     return answer
 
 
-def load_corpus(url, file):
+def question_answer(url, file, question):
     if url.strip() == '' and file == None:
         return '[ERROR]: Both URL and PDF is empty. Provide atleast one.'
     
@@ -153,16 +153,9 @@ def load_corpus(url, file):
         file_name = file_name[:-12] + file_name[-4:]
         os.rename(old_file_name, file_name)
         load_recommender(file_name)
-        
-    return 'Corpus Loaded. Now you can ask Questions.'
-
 
-def question_answer(question):
     if question.strip() == '':
         return '[ERROR]: Question field is empty'
-    
-    if not recommender.fitted:
-        return '[ERROR]: First, provide a URL or Upload a PDF and hit submit (see left panel)'
 
     return generate_answer(question)
 
@@ -174,28 +167,20 @@ with gr.Blocks() as demo:
 
     gr.Markdown(f'<center><h1>{title}</h1></center>')
     gr.Markdown(description)
-    gr.Markdown('<br>')
 
     with gr.Row():
         
         with gr.Group():
-            gr.Markdown('### Step 1: Provide URL or Upload PDF')
-            gr.Markdown('<br>')
             url = gr.Textbox(label='URL')
-            gr.Markdown("<center><h5>or<h5></center>")
+            gr.Markdown("<center><h6>or<h6></center>")
             file = gr.File(label='PDF', file_types=['.pdf'])
-            stataus = gr.Textbox(label="Output")
-            btn1 = gr.Button(value='Submit')
-            btn1.style(full_width=True)
-            btn1.click(load_corpus, inputs=[url, file], outputs=[stataus])
+            question = gr.Textbox(label='question')
+            btn = gr.Button(value='Submit')
+            btn.style(full_width=True)
 
         with gr.Group():
-            gr.Markdown('### Step 2: Now you can ask Questions:')
-            gr.Markdown('<br>')
-            question = gr.Textbox(label='question')
-            btn2 = gr.Button(value='Submit')
-            btn2.style(full_width=True)
             answer = gr.Textbox(label='answer')
-            btn2.click(question_answer, inputs=[question], outputs=[answer])
+
+        btn.click(question_answer, inputs=[url, file, question], outputs=[answer])
 
 demo.launch()