Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from nbformat.v4 import (new_notebook, new_markdown_cell, new_code_cell)
|
|
| 4 |
import pdfplumber
|
| 5 |
import google.generativeai as genai
|
| 6 |
import nbformat
|
| 7 |
-
|
| 8 |
|
| 9 |
def classify_page(statement):
|
| 10 |
genai.configure(api_key='AIzaSyBjGNru-WJoLncbVrErEzJE184MgX1o_Kg')
|
|
@@ -98,8 +98,11 @@ def classify_page(statement):
|
|
| 98 |
"""
|
| 99 |
response = chat_session.send_message(prompt)
|
| 100 |
print(response.text)
|
|
|
|
| 101 |
print(response.text.replace("```python\n", "").replace("```", "").strip())
|
| 102 |
-
|
|
|
|
|
|
|
| 103 |
|
| 104 |
def create_notebook(file, tc, bc):
|
| 105 |
notebook = new_notebook()
|
|
|
|
| 4 |
import pdfplumber
|
| 5 |
import google.generativeai as genai
|
| 6 |
import nbformat
|
| 7 |
+
import re
|
| 8 |
|
| 9 |
def classify_page(statement):
|
| 10 |
genai.configure(api_key='AIzaSyBjGNru-WJoLncbVrErEzJE184MgX1o_Kg')
|
|
|
|
| 98 |
"""
|
| 99 |
response = chat_session.send_message(prompt)
|
| 100 |
print(response.text)
|
| 101 |
+
response = response.text.replace("```python\n", "").replace("```", "").strip()
|
| 102 |
print(response.text.replace("```python\n", "").replace("```", "").strip())
|
| 103 |
+
response = re.sub(r"[^\x20-\x7E]", "", response)
|
| 104 |
+
print(response)
|
| 105 |
+
return ast.literal_eval(response)
|
| 106 |
|
| 107 |
def create_notebook(file, tc, bc):
|
| 108 |
notebook = new_notebook()
|