Upload app.py with huggingface_hub

app.py

@@ -676,6 +676,9 @@ def save_to_file(content, filename=None, prompt=None):
     elif not filename.endswith('.txt'):
         filename = f"{filename}_{generate_filename()}"
     
+    # Ensure filename has no extra whitespace
+    filename = filename.strip()
+    
     filepath = os.path.join(OUTPUT_DIR, filename)
     
     try:
@@ -692,6 +695,9 @@ def save_to_json(content, source_type, prompt, filename=None):
     if filename is None:
         filename = generate_filename(prefix=f"{source_type}_analysis", ext="json")
     
+    # Ensure filename has no extra whitespace
+    filename = filename.strip()
+    
     filepath = os.path.join(OUTPUT_DIR, filename)
     
     # Format the content from text to structured JSON
@@ -928,6 +934,9 @@ def save_to_html(content, source_type, filename=None):
     if filename is None:
         filename = generate_filename(prefix=f"{source_type}_analysis", ext="html")
     
+    # Ensure filename has no extra whitespace
+    filename = filename.strip()
+    
     filepath = os.path.join(OUTPUT_DIR, filename)
     
     try:
@@ -1582,6 +1591,8 @@ def main():
                 # Function to read a saved file
                 def read_saved_file(filename):
                     try:
+                        # Trim any whitespace from the filename
+                        filename = filename.strip()
                         filepath = os.path.join(OUTPUT_DIR, filename)
                         with open(filepath, 'r', encoding='utf-8') as f:
                             return f.read()
@@ -1601,6 +1612,8 @@ def main():
                     if not filename:
                         return None
                     try:
+                        # Trim any whitespace from the filename
+                        filename = filename.strip()
                         filepath = os.path.join(OUTPUT_DIR, filename)
                         if not os.path.exists(filepath):
                             return None
@@ -1619,6 +1632,8 @@ def main():
                         return "Please enter a filename"
                     
                     try:
+                        # Trim any whitespace from the filename
+                        filename = filename.strip()
                         filepath = os.path.join(OUTPUT_DIR, filename)
                         if not os.path.exists(filepath):
                             return "File not found"