Update app.py

app.py

@@ -1,86 +1,77 @@
 import os
-import re
 from pathlib import Path
 import json
 import base64
 from datetime import datetime
 
+# Standard imports
 import gradio as gr
-from fastapi import FastAPI, Request, Response
+from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
 import uvicorn
-import httpx
 
+# Hugging Face Spaces imports
 import spaces
 from spaces.zero.client import _get_token
 
-# Set environment variables so Gradio builds its URLs.
-os.environ["GRADIO_SSR_MODE"] = "True"
-os.environ["GRADIO_SERVER_PORT"] = "7860"
-os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
-# Tell Gradio what the node server is (host and port)
-os.environ["GRADIO_NODE_SERVER_NAME"] = "127.0.0.1:7861"
-os.environ["GRADIO_ROOT_PATH"] = "/"
-
 # Create FastAPI app
 app = FastAPI()
 
-# (Optional) Mount static files
+# Create and configure static directory
 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")
-os.environ["GRADIO_ALLOWED_PATHS"] = str(static_dir.resolve())
 
-# RESPONSE MIDDLEWARE TO FIX SSR ASSET URLS
-@app.middleware("http")
-async def fix_asset_urls(request: Request, call_next):
-    response = await call_next(request)
-    content_type = response.headers.get("content-type", "")
-    # Process only HTML responses
-    if "text/html" in content_type:
-        # Read the entire response body
-        body = b""
-        async for chunk in response.body_iterator:
-            body += chunk
-        # Step 1: Insert a slash after ':7861' if not already present.
-        fixed_body = re.sub(rb'(:7861)(?!/)', rb'\1/', body)
-        # Step 2: Fix the _appimmutable segment:
-        # Replace "_appimmutable" with "_app/immutable"
-        fixed_body = re.sub(rb'(_app)(immutable)', rb'\1/immutable', fixed_body)
-        # Step 3: Ensure that after "_app/immutable" there is a slash.
-        fixed_body = re.sub(rb'(_app/immutable)(?!/)', rb'\1/', fixed_body)
-        # Step 4: If "assets" is immediately followed by an uppercase letter, insert a slash.
-        fixed_body = re.sub(rb'(assets)(?=[A-Z])', rb'\1/', fixed_body)
-        # Build a new HTML response using the fixed body.
-        response = Response(content=fixed_body,
-                            status_code=response.status_code,
-                            headers=dict(response.headers),
-                            media_type="text/html")
-    return response
+# Tell Gradio which paths are allowed to be served
+os.environ["GRADIO_ALLOWED_PATHS"] = str(static_dir.resolve())
 
-@spaces.GPU(duration=4*60)
+@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}")
+    print(f"Token payload: {payload}")  # For debugging
+
+    # Process the text using GPU function
     processed_text = process_text(text)
-    return processed_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)
+    
+    return gr.File(value=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.Textbox(label="Output")
-    submit_btn.click(fn=process_and_save, inputs=[text_input], outputs=output)
+    output = gr.File(label="Download Processed File")
+    
+    submit_btn.click(
+        fn=process_and_save,
+        inputs=[text_input],
+        outputs=output
+    )
 
-# Mount the Gradio app into FastAPI using SSR mode (node_port 7861).
-app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True, node_port=7861)
+# 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)