Update app.py

app.py

@@ -112,11 +112,11 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         )
     
 
-    system_prompt = []
-    if prompt_template:
-        system_prompt = [{ "role": "system", "content": prompt_template }]
+   # system_prompt = []
+    #if prompt_template:
+     #   system_prompt = [{ "role": "system", "content": prompt_template }]
 
-    prompt_msg = { "role": "user", "content": prompt }
+   # prompt_msg = { "role": "user", "content": prompt }
 
 
 # The embeddings file has to be remade since the serialization is no long compatible
@@ -126,13 +126,14 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 #        new_docsearch = pickle.load(f)
         
 
-    query = str(system_prompt + history +  [prompt_msg])
+    #query = str(system_prompt + history +  [prompt_msg])
     
 #    docs = new_docsearch.similarity_search(query)
 
-    gen_ai = ChatGoogleGenerativeAI(model="gemini-pro")
+    gen_ai = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.7, top_p=0.85)
     response = gen_ai.invoke(
-        input=query,  # Pass both history and current prompt
+        input=prompt, 
+        context=history,
         #max_tokens=max_tokens,  # for open ai only
         #temperature=temperature  # for open ai only
     )