Update app.py

app.py

@@ -6,27 +6,28 @@ import google.generativeai as genai
 # Set your GEMINI API key
 api_key = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key = api_key)
+system_message = os.getenv("SYSTM_MSG")
 
-chat_model = genai.GenerativeModel(model_name='gemini-2.0-flash-exp')
-generationConfig = {temperature: 0.7, topK: 1, topP: 1, maxOutputTokens: 1024,};
-safetySettings = [
-    {
-      category: HarmCategory.HARM_CATEGORY_HARASSMENT,
-      threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
-    },
-    {
-      category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-      threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
-    },
-    {
-      category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-      threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
-    },
-    {
-      category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-      threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
-    },
-]
+model = genai.GenerativeModel(model_name="gemini-1.5-flash", system_instruction = system_message)
+# generationConfig = GenerationConfig(max_output_tokens=2048, temperature=0.7, top_p=1, top_k=32)
+# safetySettings = [
+#     {
+#       category: HarmCategory.HARM_CATEGORY_HARASSMENT,
+#       threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+#     },
+#     {
+#       category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+#       threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+#     },
+#     {
+#       category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+#       threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+#     },
+#     {
+#       category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+#       threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+#     },
+# ]
 
 # query = "Tell me about yourself"
 # response = model.generate_content(f"""You are an friendly assistant.Answer the following with high precision.{query}""")display(Markdown(response.text))
@@ -38,13 +39,10 @@ new_year_levels = [os.getenv("First_Word"), os.getenv("Second_Word"), os.getenv(
 def ask_openai(prompt, secret_word):
     """
     Sends the user prompt and safeguards to the OpenAI API to generate a response.
-    """
-    system_message = os.getenv("SYSTM_MSG")
-    
+    """    
     try:
-        chat = model.start_chat(generationConfig, safetySettings,
-                                history=[{"role": "system", "content": system_message}, {"role": "user", "content": prompt},])
-        
+        # chat = model.start_chat(history=[{"role": "user", "content": prompt},])
+        response = model.generate_content(prompt)
         print(repsonse)
         return response.text