Update app.py

app.py

@@ -97,9 +97,9 @@ def classify_page(statement):
     Input: {statement}
     """
     response = chat_session.send_message(prompt)
+    print(response.text)
     print(response.text.replace("```python\n", "").replace("```", "").strip())
-    # return ast.literal_eval(response.text.replace("```python\n", "").replace("```", "").strip())
-    return response.text.replace("```python\n", "").replace("```", "").strip()
+    return ast.literal_eval(response.text.replace("```python\n", "").replace("```", "").strip())
 
 def create_notebook(file, tc, bc):
     notebook = new_notebook()