Update app.py

app.py

@@ -239,25 +239,19 @@ title = 'PDF GPT Turbo'
 description = """ PDF GPT Turbo allows you to chat with your PDF files. It uses Google's Universal Sentence Encoder with Deep averaging network (DAN) to give hallucination free response by improving the embedding quality of OpenAI. It cites the page number in square brackets([Page No.]) and shows where the information is located, adding credibility to the responses."""
 
 with gr.Blocks(css="""#chatbot { font-size: 14px; min-height: 1200; }""") as demo:
-
     gr.Markdown(f'<center><h3>{title}</h3></center>')
     gr.Markdown(description)
 
     with gr.Row():
-        
-        with gr.Group():
-            gr.Markdown(f'<p style="text-align:center">Get your Open AI API key <a href="https://platform.openai.com/account/api-keys">here</a></p>')
-            with gr.Accordion("API Key"):
-                openAI_key = gr.Textbox(label='Enter your OpenAI API key here')
-                url = gr.Textbox(label='Enter PDF URL here   (Example: https://arxiv.org/pdf/1706.03762.pdf )')
+        with gr.Column():
+            # API Key and File Inputs
+            with gr.Accordion("API Key and PDF"):
+                openAI_key = gr.Textbox(label='Enter your OpenAI API key here', type='password')
+                url = gr.Textbox(label='Enter PDF URL here (Example: https://arxiv.org/pdf/1706.03762.pdf)')
                 gr.Markdown("<center><h4>OR<h4></center>")
-                file = gr.File(label='Upload your PDF/ Research Paper / Book here', file_types=['.pdf'])
-            question = gr.Textbox(label='Enter your question here')
-            gr.Examples(
-                [[q] for q in questions],
-                inputs=[question],
-                label="PRE-DEFINED QUESTIONS: Click on a question to auto-fill the input box, then press Enter!",
-            )
+                file = gr.File(label='Upload your PDF/Research Paper/Book here', file_types=['.pdf'])
+            
+            # Model Selection
             model = gr.Radio(
                 choices=[
                     'gpt-3.5-turbo', 
@@ -271,18 +265,60 @@ with gr.Blocks(css="""#chatbot { font-size: 14px; min-height: 1200; }""") as dem
                 label='Select Model', 
                 value='gpt-3.5-turbo'
             )
-            btn = gr.Button(value='Submit')
 
-        with gr.Group():
-            chatbot = gr.Chatbot(placeholder="Chat History", label="Chat History", elem_id="chatbot")
+            # Chat Interface
+            chatbot = gr.Chatbot(label="Chat History", type="messages")
+            msg = gr.Textbox(label="Enter your question here", lines=2)
+            clear = gr.ClearButton([msg, chatbot])
 
+            # Example Questions
+            gr.Examples(
+                [[q] for q in questions],
+                inputs=[msg],
+                label="PRE-DEFINED QUESTIONS: Click on a question to auto-fill the input box",
+            )
 
-#
-    # Bind the click event of the button to the question_answer function
-    btn.click(
-        question_answer,
-        inputs=[chatbot, url, file, question, openAI_key, model],
-        outputs=[chatbot],
+    def respond(message, chat_history, url_value, file_value, key_value, model_value):
+        try:
+            if key_value.strip() == '':
+                return chat_history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": '[ERROR]: Please enter your OpenAI API key'}
+                ]
+            
+            if url_value.strip() == '' and file_value is None:
+                return chat_history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": '[ERROR]: Both URL and PDF are empty. Provide at least one'}
+                ]
+
+            # Process PDF and generate answer
+            if url_value.strip() != '':
+                download_pdf(url_value, 'corpus.pdf')
+                load_recommender('corpus.pdf')
+            else:
+                old_file_name = file_value.name
+                file_name = old_file_name[:-12] + old_file_name[-4:]
+                os.rename(old_file_name, file_name)
+                load_recommender(file_name)
+
+            answer = generate_answer(message, key_value, model_value)
+            
+            return chat_history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": answer}
+            ]
+
+        except Exception as e:
+            return chat_history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": f'[ERROR]: {str(e)}'}
+            ]
+
+    msg.submit(
+        respond, 
+        [msg, chatbot, url, file, openAI_key, model], 
+        [chatbot]
     )
 
 demo.launch()