add application files

app.py

@@ -1,6 +1,17 @@
+import os
 import gradio as gr
 from model.controller import Controller
 
+# Change working directory to /home/user/app
+os.chdir("/home/user/app")
+
+# Download and prepare data files in the correct directory
+os.system('wget -O processed_cases.csv "https://drive.usercontent.google.com/download?id=1jMuQtywo0mbj7ZHCCsyE8xurbSyVVCst&export=download&confirm=t&uuid=2f681c98-86f8-4159-9e03-673cdcbc7cb51"')
+os.system('wget -O chromadb_collection.rar "https://drive.usercontent.google.com/download?id=1-3N-QjK6QMogy2mcNEKjTg_9yIrSjh4A&export=download&confirm=t&uuid=ceed27ad-0f81-4c16-b70f-68cebcc55279"')
+os.system('unrar x chromadb_collection.rar')
+os.system('mv content/chromadb_collections chromadb_collections')
+os.system('rm -r content')
+
 # Initialize your controller
 bot = Controller()
 
@@ -13,7 +24,7 @@ with gr.Blocks() as interface:
     gr.Markdown("## RAG Law Chatbot")
     
     chatbot = gr.Chatbot()
-    user_input = gr.Textbox(show_label=False, placeholder="Enter your law question...")  # Removed .style(container=False)
+    user_input = gr.Textbox(show_label=False, placeholder="Enter your law question...")
     send_button = gr.Button("Send")
 
     def chat_update(user_message, history):

model/chat.py

@@ -17,7 +17,8 @@ class Chat:
 
         info_list = self.rag_handler.get_information(message)
         prompt = self.prompt_handler.get_prompt(message, info_list)
-        response = self.llm.get_LLM_response(prompt=prompt)
+        
+        # response = self.llm.get_LLM_response(prompt=prompt)
 
-        self.response_history.append(response)
-        return response
+        # self.response_history.append(response)
+        return prompt