Update main.py

main.py

@@ -1,102 +1,61 @@
 import os
 from pathlib import Path
-import json
-import base64
+import json, base64, time
 from datetime import datetime
-import time 
 
-
-# Standard imports
 import gradio as gr
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 import uvicorn
 
-# Hugging Face Spaces imports
 import spaces
 from spaces.zero.client import _get_token
 
-# Create FastAPI app
 app = FastAPI()
 
-# Create and configure static directory
+# Create and mount a static directory if needed
 static_dir = Path("./static")
 static_dir.mkdir(parents=True, exist_ok=True)
-
-# Mount static directory to FastAPI
 app.mount("/static", StaticFiles(directory="static"), name="static")
-#app.mount("/_app/immutable/assets", StaticFiles(directory="assets"), name="immutable_assets")
-
-# Tell Gradio which paths are allowed to be served
-os.environ["GRADIO_ALLOWED_PATHS"] = str(static_dir.resolve())
 
+# Remove or override the default GRADIO_ROOT_PATH
+os.environ["GRADIO_ROOT_PATH"] = "/assets"
 
-@spaces.GPU(duration=4*60)  # Specify GPU duration in seconds, added for testing
+@spaces.GPU(duration=4*60)
 def process_text(text):
-    """Example GPU function - in reality, this might be model inference"""
     time.sleep(10)
-
     return text.upper()
-# I've tried also wrapping the immediate function called by gradio, and the zerogpu = True trick
 
 def process_and_save(request: gr.Request, text):
-    """Main processing function that handles tokens and calls GPU function"""
-    # Get and decode the authentication token
     token = _get_token(request)
     payload = token.split('.')[1]
     payload = f"{payload}{'=' * ((4 - len(payload) % 4) % 4)}"
     payload = json.loads(base64.urlsafe_b64decode(payload).decode())
-    print(f"Token payload: {payload}")  # For debugging
+    print(f"Token payload: {payload}")
 
-    # Process the text using GPU function
     processed_text = process_text(text)
-    
-    # Save to file
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     file_path = static_dir / f"output_{timestamp}.txt"
     with open(file_path, "w") as f:
         f.write(processed_text)
-    
-    # Convert file_path to a string before returning
     return gr.File(value=str(file_path))
 
-# Mark main function as not requiring GPU
 process_and_save.zerogpu = True
 
-# Create Gradio interface
 with gr.Blocks() as demo:
     text_input = gr.Textbox(label="Enter some text")
     submit_btn = gr.Button("Process and Download")
     output = gr.File(label="Download Processed File")
-    
-    submit_btn.click(
-        fn=process_and_save,
-        inputs=[text_input],
-        outputs=output
-    )
-
-
-#os.environ["GRADIO_ROOT_PATH"] = "https://m7n-zero-gpu-uvicorn-ssr-example.hf.space/_app/immutable/"
 
-os.environ["GRADIO_ROOT_PATH"] = "/_app/immutable"
+    submit_btn.click(fn=process_and_save, inputs=[text_input], outputs=output)
 
-# Mount Gradio app to FastAPI with SSR mode for Spaces
+# Mount Gradio app with SSR
 app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True)
-#app.mount("/_app/immutable/assets", StaticFiles(directory="assets"), name="immutable_assets")
-
-
-from pathlib import Path
 
+# Make sure assets are served from /assets
 assets_dir = Path("./assets")
-if not assets_dir.exists():
-    assets_dir.mkdir(parents=True, exist_ok=True)  # Create the folder if it doesn't exist
+assets_dir.mkdir(parents=True, exist_ok=True)
+app.mount("/assets", StaticFiles(directory=str(assets_dir)), name="assets")
 
-app.mount("/_app/immutable/assets", StaticFiles(directory=str(assets_dir)), name="immutable_assets")
-
-
-# Run server
 if __name__ == "__main__":
-    # Set SSR mode for Spaces
-    
-
     uvicorn.run(app, host="0.0.0.0", port=7860)