Update app.py

app.py

@@ -263,12 +263,16 @@ with gr.Blocks() as demo:
                 label='Select Model', 
                 value='gpt-4o-mini'
             )
-
+            
             # Chat Interface
-            chatbot = gr.Chatbot(label="Chat History", type="messages")
+            chatbot = gr.Chatbot(label="Chat History")  # default: list of [user, bot] pairs
             msg = gr.Textbox(label="Enter your question here", lines=2)
             submit_btn = gr.Button("Submit")
             clear = gr.ClearButton([msg, chatbot])
+            #chatbot = gr.Chatbot(label="Chat History", type="messages")
+            #msg = gr.Textbox(label="Enter your question here", lines=2)
+            #submit_btn = gr.Button("Submit")
+            #clear = gr.ClearButton([msg, chatbot])
 
             # Example Questions
             gr.Examples(
@@ -277,52 +281,50 @@ with gr.Blocks() as demo:
                 label="PRE-DEFINED QUESTIONS: Click on a question to auto-fill the input box",
             )
 
-    def respond(message, chat_history, url_value, file_value, key_value, model_value):
-        if message.strip() == "":
-            return "", chat_history  # Return empty message if no input
+def respond(message, chat_history, url_value, file_value, key_value, model_value):
+    if message.strip() == "":
+        return "", chat_history  # nothing to do
 
-        try:
-            # Ensure chat_history is initialized properly
-            if chat_history is None:
-                chat_history = []
+    # Initialize chat_history as a list of [user, assistant] pairs
+    if chat_history is None:
+        chat_history = []
 
-            if key_value.strip() == '':
-                chat_history.append({"role": "user", "content": message})
-                chat_history.append({"role": "assistant", "content": '[ERROR]: Please enter your OpenAI API key'})
-                return "", chat_history
-
-            if url_value.strip() == '' and file_value is None:
-                chat_history.append({"role": "user", "content": message})
-                chat_history.append({"role": "assistant", "content": '[ERROR]: Both URL and PDF are empty. Provide at least one'})
-                return "", chat_history
+    try:
+        # Basic validations
+        if key_value.strip() == '':
+            chat_history.append([message, '[ERROR]: Please enter your OpenAI API key'])
+            return "", chat_history
 
-            # 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)
+        if url_value.strip() == '' and file_value is None:
+            chat_history.append([message, '[ERROR]: Both URL and PDF are empty. Provide at least one'])
+            return "", chat_history
 
-            answer = generate_answer(message, key_value, model_value)
+        # Prepare corpus (URL or uploaded file)
+        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)
 
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": answer})
+        # Generate answer
+        answer = generate_answer(message, key_value, model_value)
 
-            return "", chat_history
+        # Append pair [user, assistant]
+        chat_history.append([message, answer])
+        return "", chat_history
 
-        except Exception as e:
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": f'[ERROR]: {str(e)}'})
-            return "", chat_history
+    except Exception as e:
+        chat_history.append([message, f'[ERROR]: {str(e)}'])
+        return "", chat_history
 
     submit_btn.click(
-        respond, 
-        [msg, chatbot, url, file, openAI_key, model], 
-        [msg, chatbot]
-    )
+    respond, 
+    [msg, chatbot, url, file, openAI_key, model], 
+    [msg, chatbot]
+)
 
     msg.submit(
         respond,