Update app.py

app.py

@@ -783,7 +783,7 @@ Standalone question:""")
 
 
 
-def submit_message_OLD(prompt, prompt_template, temperature, max_tokens, context_length, state):
+def submit_message(prompt, prompt_template, temperature, max_tokens, context_length, state):
     
     
     history = state['messages']
@@ -793,38 +793,8 @@ def submit_message_OLD(prompt, prompt_template, temperature, max_tokens, context
     
     prompt_template = prompt_templates[prompt_template]
 
-    with open("prompts_archive.csv", "a") as csvfile:
-        writer = csv.DictWriter(csvfile, fieldnames=["prompt", "time"])
-        writer.writerow(
-            {"prompt": str(prompt), "time": str(datetime.now())}
-        )
-    
-   # with open(prompts_archive_file, "a") as csvfile:
-   #     writer = csv.DictWriter(csvfile, fieldnames=["prompt", "time"])
-   #     writer.writerow(
-   #         {"prompt": str(prompt), "time": str(datetime.now())}
-   #     )
-   # commit_url = repo.push_to_hub()
-   # print(commit_url)
-
-    system_prompt = []
-    if prompt_template:
-        system_prompt = [{ "role": "system", "content": prompt_template }]
-
-    prompt_msg = { "role": "user", "content": prompt }
 
-
-    #try:
-
-    with open("embeddings.pkl", 'rb') as f: 
-        new_docsearch = pickle.load(f)
-        
-
-    query = str(system_prompt + history +  [prompt_msg])
-    
-    docs = new_docsearch.similarity_search(query)
-
-    chain = load_qa_chain(ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff")
+    #chain = load_qa_chain(ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff")
     #completion = chain.run(input_documents=docs, question=query)
     
 
@@ -835,20 +805,8 @@ def submit_message_OLD(prompt, prompt_template, temperature, max_tokens, context
     
     completion = { "content": completion }
     
-    
-    #state['total_tokens'] += completion['usage']['total_tokens']
-
 
-    
-    #except Exception as e:
-    #    history.append(prompt_msg.copy())
-    #    error  = {
-    #        "role": "system",
-    #        "content": f"Error: {e}"
-    #    }
-    #    history.append(error.copy())
 
-    #total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
 
     chat_messages = [(prompt_msg['content'], completion['content'])]    
     return '', chat_messages,  state    # total_tokens_used_msg,