Update app.py

app.py

@@ -1,8 +1,9 @@
 import os
 # Disable GPU to avoid CUDA errors
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
+os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"  # Prevent TensorFlow memory issues
 # Suppress TensorFlow warnings
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # More aggressive suppression
 
 import gradio as gr
 import torch
@@ -763,9 +764,13 @@ def create_gradio_interface():
 # --- Application Startup ---
 def run_application():
     try:
+        logger.info("🧬 Initializing Gene Analysis Pipeline...")
         main_gradio_app = create_gradio_interface()
+        if main_gradio_app is None:
+            raise RuntimeError("Gradio interface creation returned None")
+        logger.info("✅ Gradio interface created successfully")
         main_gradio_app = gr.mount_gradio_app(app, main_gradio_app, path="/gradio")
-        logger.info("🧬 Gene Analysis Pipeline Starting...")
+        logger.info("✅ Gradio mounted to FastAPI at /gradio")
         logger.info("=" * 50)
         logger.info("🔍 Checking system components...")
         logger.info(f"🤖 Boundary Model: {'✅ Loaded' if boundary_model else '❌ Missing'}")
@@ -784,22 +789,28 @@ def run_application():
             host="0.0.0.0",
             port=7860,
             log_level="info",
-            access_log=True
+            access_log=True,
+            timeout_keep_alive=120
         )
     except Exception as main_error:
         logger.error(f"Application startup failed: {main_error}", exc_info=True)
         try:
             logger.info("🔄 Falling back to Gradio-only mode...")
             fallback_gradio_app = create_gradio_interface()
+            if fallback_gradio_app is None:
+                raise RuntimeError("Fallback Gradio interface creation returned None")
+            logger.info("✅ Fallback Gradio interface created successfully")
             logger.info("🧬 Gradio interface available at: http://localhost:7860")
             fallback_gradio_app.launch(
                 server_name="0.0.0.0",
                 server_port=7860,
-                prevent_thread_lock=True
+                prevent_thread_lock=True,
+                quiet=True
             )
         except Exception as fallback_error:
             logger.error(f"Fallback failed: {fallback_error}", exc_info=True)
-            print("❌ Application failed to start. Check logs for details.")
+            print("❌ Application failed to start. Check logs at /tmp/app.log for details.")
+            sys.exit(1)
 
 if __name__ == "__main__":
     print("🧬 Gene Analysis Pipeline Starting...")