Update app.py

app.py

@@ -9,6 +9,7 @@ from i_search import i_search as i_s
 from datetime import datetime
 import logging
 import json
+import nltk  # Import nltk for sentence tokenization
 
 now = datetime.now()
 date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
@@ -24,7 +25,7 @@ logging.basicConfig(
     format="%(asctime)s - %(levelname)s - %(message)s",
 )
 
-agents =[
+agents = [
     "WEB_DEV",
     "AI_SYSTEM_PROMPT",
     "PYTHON_CODE_DEV"
@@ -478,7 +479,7 @@ def project_explorer(path):
     tree = get_file_tree(path)
     display_file_tree(tree)
 
-def chat_app_logic(message, history):
+def chat_app_logic(message, history, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
     # Your existing code here
     
     try:
@@ -556,8 +557,8 @@ def main():
                 gr.Button(value=example[0]).click(lambda: chat_app_logic, inputs=[(example[0], message, purpose)], outputs=chatbot)
 
             # Connect components to the chat app logic
-            submit_button.click(chat_app_logic, inputs=[message, history], outputs=chatbot)
-            message.submit(chat_app_logic, inputs=[ history, message], outputs=chatbot)
+            submit_button.click(chat_app_logic, inputs=[message, history, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty], outputs=chatbot)
+            message.submit(chat_app_logic, inputs=[ history, message, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty], outputs=chatbot)
         # Connect components to the project explorer
         explore_button.click(project_explorer, inputs=[project_path], outputs=project_output)