Update app_gpu.py

app_gpu.py

@@ -1,5 +1,5 @@
 
-# import spaces
+import spaces
 import os
 import openai
 from dotenv import load_dotenv
@@ -66,7 +66,7 @@ class cbfs:
             return_generated_question=True,
         )
 
-    # @spaces.GPU
+    @spaces.GPU
     def convchain(self, query):
         if not query:
             return [("User", ""), ("ChatBot", "")]
@@ -94,18 +94,17 @@ def initialize_cbfs(db_choice, model_choice):
     else:
         return None
 
-def chat_history(query, db_choice, model_choice, cb):
-# def chat_history(query, db_choice, model_choice):
+def chat_history(query, db_choice, model_choice):
     """Handles chat submissions. Reminds the user to select a document if none is selected."""
-    # cb = initialize_cbfs(db_choice, model_choice)  # Reinitialize cbfs 
+    cb = initialize_cbfs(db_choice, model_choice)  # Reinitialize cbfs 
     if cb is None:  # If cb is not initialized, remind to select a document
         return [("ChatBot", "Please select a document from the dropdown menu before submitting your query.")], ""
     else:
         return cb.convchain(query), ""  # Clear input box by returning empty string
 
-def clear_history(cb):
-# def clear_history(db_choice, model_choice):
-    # cb = initialize_cbfs(db_choice, model_choice)  # Reinitialize cbfs to clear history
+
+def clear_history(db_choice, model_choice):
+    cb = initialize_cbfs(db_choice, model_choice)  # Reinitialize cbfs to clear history
     if cb is None:  # Check if cbfs instance is None
         return [], ""  # No error message, simply clear the UI components
     else:
@@ -159,13 +158,10 @@ with gr.Blocks() as demo:
     )
 
     # Define interactions for both submit button and Enter key
-    inp.submit(fn=chat_history, inputs=[inp, db_choice, model_choice, cbfs_instance], outputs=[output, inp])
-    button_submit.click(fn=chat_history, inputs=[inp, db_choice, model_choice, cbfs_instance], outputs=[output, inp])
-    button_clearhistory.click(fn=clear_history, inputs=cbfs_instance, outputs=[output, inp])
 
-    # inp.submit(fn=chat_history, inputs=[inp, db_choice, model_choice], outputs=[output, inp])
-    # button_submit.click(fn=chat_history, inputs=[inp, db_choice, model_choice], outputs=[output, inp])
-    # button_clearhistory.click(fn=clear_history, inputs=[db_choice, model_choice], outputs=[output, inp])
+    inp.submit(fn=chat_history, inputs=[inp, db_choice, model_choice], outputs=[output, inp])
+    button_submit.click(fn=chat_history, inputs=[inp, db_choice, model_choice], outputs=[output, inp])
+    button_clearhistory.click(fn=clear_history, inputs=[db_choice, model_choice], outputs=[output, inp])
 
 
 # Launch the Gradio app