Spaces:
Runtime error
Runtime error
Commit ·
2c5fe24
1
Parent(s): 9f7c458
add token debugging6
Browse files- app.py +30 -51
- requirements.txt +3 -1
app.py
CHANGED
|
@@ -1,56 +1,35 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
import pandas as pd
|
| 3 |
-
from modal import Function
|
| 4 |
import os
|
| 5 |
-
import
|
|
|
|
| 6 |
|
| 7 |
-
def
|
| 8 |
-
"""Initialize Modal with token
|
| 9 |
try:
|
| 10 |
-
# Check if token exists in environment
|
| 11 |
token = os.environ.get('MODAL_TOKEN')
|
| 12 |
if not token:
|
| 13 |
st.error("MODAL_TOKEN not found in environment variables")
|
| 14 |
-
|
| 15 |
-
return False
|
| 16 |
|
| 17 |
-
# Create
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
# Create directory with proper permissions
|
| 22 |
-
os.makedirs(token_dir, mode=0o700, exist_ok=True)
|
| 23 |
-
|
| 24 |
-
# Write token file with proper permissions
|
| 25 |
-
with open(token_file, "w") as f:
|
| 26 |
-
f.write(token)
|
| 27 |
-
|
| 28 |
-
# Set file permissions to be readable only by owner
|
| 29 |
-
os.chmod(token_file, stat.S_IRUSR | stat.S_IWUSR)
|
| 30 |
-
|
| 31 |
-
# Verify the token file exists
|
| 32 |
-
if not os.path.exists(token_file):
|
| 33 |
-
st.error("Failed to create token file")
|
| 34 |
-
return False
|
| 35 |
-
|
| 36 |
-
st.success("Modal token configured successfully")
|
| 37 |
-
|
| 38 |
-
# Debug info
|
| 39 |
-
if st.checkbox("Show Token File Info"):
|
| 40 |
-
file_stat = os.stat(token_file)
|
| 41 |
-
st.write({
|
| 42 |
-
"Token file": token_file,
|
| 43 |
-
"Exists": os.path.exists(token_file),
|
| 44 |
-
"Permissions": oct(file_stat.st_mode)[-3:],
|
| 45 |
-
"Owner": file_stat.st_uid,
|
| 46 |
-
"Group": file_stat.st_gid
|
| 47 |
-
})
|
| 48 |
-
|
| 49 |
-
return True
|
| 50 |
except Exception as e:
|
| 51 |
-
st.error(f"Failed to initialize Modal: {str(e)}")
|
| 52 |
-
st.exception(e)
|
| 53 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
|
| 55 |
def main():
|
| 56 |
st.title("Financial Statement Analyzer")
|
|
@@ -60,14 +39,15 @@ def main():
|
|
| 60 |
st.write("Environment Variables:")
|
| 61 |
st.write({k: "***" if k == "MODAL_TOKEN" else v
|
| 62 |
for k, v in os.environ.items()})
|
| 63 |
-
st.write("\nCurrent Working Directory:", os.getcwd())
|
| 64 |
-
st.write("User Home:", os.path.expanduser("~"))
|
| 65 |
-
st.write("User:", os.getenv("USER"))
|
| 66 |
|
| 67 |
-
# Initialize Modal
|
| 68 |
-
|
|
|
|
|
|
|
| 69 |
return
|
| 70 |
|
|
|
|
|
|
|
| 71 |
uploaded_files = st.file_uploader(
|
| 72 |
"Choose PDF files",
|
| 73 |
type="pdf",
|
|
@@ -86,8 +66,7 @@ def main():
|
|
| 86 |
progress_bar.progress(25)
|
| 87 |
|
| 88 |
# Process PDF through Modal backend
|
| 89 |
-
|
| 90 |
-
financial_data = pdf_processor.remote(file)
|
| 91 |
progress_bar.progress(75)
|
| 92 |
|
| 93 |
if financial_data:
|
|
@@ -108,7 +87,7 @@ def main():
|
|
| 108 |
|
| 109 |
except Exception as e:
|
| 110 |
st.error("Error during processing")
|
| 111 |
-
st.exception(e)
|
| 112 |
progress_bar.empty()
|
| 113 |
|
| 114 |
if __name__ == "__main__":
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
import pandas as pd
|
|
|
|
| 3 |
import os
|
| 4 |
+
from modal import Client, Function
|
| 5 |
+
import asyncio
|
| 6 |
|
| 7 |
+
def get_modal_client():
|
| 8 |
+
"""Initialize Modal client with token"""
|
| 9 |
try:
|
|
|
|
| 10 |
token = os.environ.get('MODAL_TOKEN')
|
| 11 |
if not token:
|
| 12 |
st.error("MODAL_TOKEN not found in environment variables")
|
| 13 |
+
return None
|
|
|
|
| 14 |
|
| 15 |
+
# Create Modal client with token
|
| 16 |
+
client = asyncio.run(Client.from_token(token))
|
| 17 |
+
return client
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
except Exception as e:
|
| 19 |
+
st.error(f"Failed to initialize Modal client: {str(e)}")
|
| 20 |
+
st.exception(e)
|
| 21 |
+
return None
|
| 22 |
+
|
| 23 |
+
def process_pdf_with_modal(file, client):
|
| 24 |
+
"""Process PDF using Modal with specific client"""
|
| 25 |
+
try:
|
| 26 |
+
# Use the client to get the function
|
| 27 |
+
pdf_processor = Function.lookup("stem", "process_pdf", client=client)
|
| 28 |
+
return pdf_processor.remote(file)
|
| 29 |
+
except Exception as e:
|
| 30 |
+
st.error(f"Error in Modal processing: {str(e)}")
|
| 31 |
+
st.exception(e)
|
| 32 |
+
return None
|
| 33 |
|
| 34 |
def main():
|
| 35 |
st.title("Financial Statement Analyzer")
|
|
|
|
| 39 |
st.write("Environment Variables:")
|
| 40 |
st.write({k: "***" if k == "MODAL_TOKEN" else v
|
| 41 |
for k, v in os.environ.items()})
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
+
# Initialize Modal client
|
| 44 |
+
client = get_modal_client()
|
| 45 |
+
if client is None:
|
| 46 |
+
st.error("Could not initialize Modal client")
|
| 47 |
return
|
| 48 |
|
| 49 |
+
st.success("Modal client initialized successfully")
|
| 50 |
+
|
| 51 |
uploaded_files = st.file_uploader(
|
| 52 |
"Choose PDF files",
|
| 53 |
type="pdf",
|
|
|
|
| 66 |
progress_bar.progress(25)
|
| 67 |
|
| 68 |
# Process PDF through Modal backend
|
| 69 |
+
financial_data = process_pdf_with_modal(file, client)
|
|
|
|
| 70 |
progress_bar.progress(75)
|
| 71 |
|
| 72 |
if financial_data:
|
|
|
|
| 87 |
|
| 88 |
except Exception as e:
|
| 89 |
st.error("Error during processing")
|
| 90 |
+
st.exception(e)
|
| 91 |
progress_bar.empty()
|
| 92 |
|
| 93 |
if __name__ == "__main__":
|
requirements.txt
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
modal
|
| 2 |
streamlit>=1.32.0
|
| 3 |
pandas>=2.2.0
|
| 4 |
-
python-dotenv>=1.0.0
|
|
|
|
|
|
|
|
|
| 1 |
modal
|
| 2 |
streamlit>=1.32.0
|
| 3 |
pandas>=2.2.0
|
| 4 |
+
python-dotenv>=1.0.0
|
| 5 |
+
asyncio
|
| 6 |
+
aiohttp
|