Update app.py

app.py

@@ -42,15 +42,20 @@ def generate_reaction_steps(reactants, products):
     
     chat_session = model.start_chat(history=[])
     response = chat_session.send_message(prompt)
-    print(response)
-    # Extract the JSON output
+    
+    # Extract the JSON content from the response
     try:
-        steps = json.loads(response.text)
-    except json.JSONDecodeError:
+        # Parsing the raw response to extract the JSON text
+        content = response.result["candidates"][0]["content"]["parts"][0]["text"]
+        
+        # Loading the JSON string to a Python dictionary
+        steps = json.loads(content)
+    except (json.JSONDecodeError, KeyError):
         steps = {"error": "Failed to decode JSON from Gemini response."}
     
     return steps
 
+
 # Gradio interface
 def process_reaction(reactants, products):
     steps = generate_reaction_steps(reactants, products)