added 0gpu

app.py

@@ -1,10 +1,19 @@
 import os
 from pathlib import Path
+import json
+import base64
+from datetime import datetime
+
+# Standard imports
 import gradio as gr
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 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()
 
@@ -18,34 +27,51 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 # Tell Gradio which paths are allowed to be served
 os.environ["GRADIO_ALLOWED_PATHS"] = str(static_dir.resolve())
 
-def process_and_save(text):
-    """Simple function that saves text to a file and returns file path"""
-    # Save text to a file in static directory
-    file_path = static_dir / "output.txt"
+@spaces.GPU(duration=60)  # Specify GPU duration in seconds
+def process_text(text):
+    """Example GPU function - in reality, this might be model inference"""
+    return text.upper()
+
+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
+
+    # 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(text)
+        f.write(processed_text)
     
-    # Return file path for download
     return gr.File(value=file_path)
 
-# Mark function as not requiring GPU
+# 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("Save and Download")
-    output = gr.File(label="Download File")
+    submit_btn = gr.Button("Process and Download")
+    output = gr.File(label="Download Processed File")
     
     submit_btn.click(
         fn=process_and_save,
-        inputs=text_input,
+        inputs=[text_input],
         outputs=output
     )
 
-# Mount Gradio app to FastAPI
-app = gr.mount_gradio_app(app, demo, path="/")
+# Mount Gradio app to FastAPI with SSR mode for Spaces
+app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True)
 
 # Run server
 if __name__ == "__main__":
+    # Set SSR mode for Spaces
+    os.environ["GRADIO_SSR_MODE"] = "True"
     uvicorn.run(app, host="0.0.0.0", port=7860)