Upload app.py with huggingface_hub

app.py

@@ -20,10 +20,20 @@ import tempfile
 import logging
 import traceback
 
-# Set up logging
-LOG_DIR = "logs"
-os.makedirs(LOG_DIR, exist_ok=True)
-log_file = os.path.join(LOG_DIR, f"app_debug_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.log")
+# Constants
+IMAGENET_MEAN = (0.485, 0.456, 0.406)
+IMAGENET_STD = (0.229, 0.224, 0.225)
+
+# Configuration
+MODEL_NAME = "OpenGVLab/InternVL2_5-8B"  # Smaller model for faster loading
+IMAGE_SIZE = 448
+OUTPUT_DIR = "saved_outputs"  # Changed to a visible repo directory
+
+# Create output directory if it doesn't exist
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+# Set up logging to write to saved_outputs directory
+log_file = os.path.join(OUTPUT_DIR, f"debug_log_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.log")
 
 # Configure logging
 logging.basicConfig(
@@ -83,18 +93,6 @@ def log_tensor_info(tensor, name="tensor"):
     except Exception as e:
         logger.error(f"Error logging tensor info for {name}: {str(e)}")
 
-# Constants
-IMAGENET_MEAN = (0.485, 0.456, 0.406)
-IMAGENET_STD = (0.229, 0.224, 0.225)
-
-# Configuration
-MODEL_NAME = "OpenGVLab/InternVL2_5-8B"  # Smaller model for faster loading
-IMAGE_SIZE = 448
-OUTPUT_DIR = "saved_outputs"  # Changed to a visible repo directory
-
-# Create output directory if it doesn't exist
-os.makedirs(OUTPUT_DIR, exist_ok=True)
-
 # Set up environment variables
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 
@@ -1410,6 +1408,23 @@ def process_image_with_text(image, prompt):
         logger.error(traceback.format_exc())
         return f"Error processing image: {str(e)}"
 
+# Function to get log file content
+def get_latest_log_content():
+    """Get the content of the latest log file for display in the UI."""
+    try:
+        log_files = sorted(glob.glob(os.path.join(OUTPUT_DIR, "debug_log_*.log")))
+        if not log_files:
+            return "No log files found."
+        
+        latest_log = log_files[-1]
+        with open(latest_log, 'r') as f:
+            # Get the last 100 lines (most recent logs)
+            lines = f.readlines()
+            last_lines = lines[-100:] if len(lines) > 100 else lines
+            return "".join(last_lines)
+    except Exception as e:
+        return f"Error reading log file: {str(e)}"
+
 # Main function
 def main():
     # Load the model
@@ -1538,6 +1553,27 @@ def main():
                     outputs=save_status_folder
                 )
             
+            # Add a new tab for logs
+            with gr.Tab("Debug Logs"):
+                gr.Markdown("## Debug Logs")
+                gr.Markdown("View the latest application logs to diagnose issues:")
+                
+                log_output = gr.Textbox(
+                    value=get_latest_log_content(),
+                    lines=20,
+                    label="Latest Log Entries",
+                    elem_id="log_output"
+                )
+                
+                refresh_btn = gr.Button("Refresh Logs")
+                refresh_btn.click(
+                    fn=get_latest_log_content,
+                    inputs=[],
+                    outputs=[log_output]
+                )
+                
+                gr.Markdown("**Note**: Log files are also saved to the `saved_outputs` directory and can be accessed via the Files tab.")
+            
             # New tab for saved output files
             with gr.Tab("Saved Outputs"):
                 refresh_btn = gr.Button("Refresh File List")