Update app.py

app.py

@@ -105,16 +105,16 @@ def main(api_key, user_query):
     try:
         # Initialize the Google Gemini model with the provided API key
         genai.configure(api_key=api_key)
-        # generation_config = {
-        #     "temperature": 0.2,
-        #     "top_p": 0.95,
-        #     "top_k": 64,
-        #     "max_output_tokens": 8192,
-        #     "response_mime_type": "text/plain",
-        # }
+        generation_config = {
+            # "temperature": 0.2,
+            # "top_p": 0.95,
+            # "top_k": 64,
+            # "max_output_tokens": 8192,
+            "response_mime_type": "application/json",
+        }
         model = genai.GenerativeModel(
             model_name="gemini-1.5-flash-exp-0827",
-            # generation_config=generation_config,
+            generation_config=generation_config,
         )
     except Exception as e:
         return f"Failed to initialize Google Gemini model. Error: {str(e)}", ""