Update main.py

main.py

@@ -19,7 +19,7 @@ app = FastAPI()
 
 # Set up upload folder and allowed extensions
 UPLOAD_FOLDER = 'uploads'
-ALLOWED_EXTENSIONS = {'pdf', 'docx', 'png', 'jpg', 'jpeg'}
+ALLOWED_EXTENSIONS = {'pdf', 'docx', 'txt'}
 MAX_CONTENT_LENGTH = 16 * 1024 * 1024  # 16 MB
 
 if not os.path.exists(UPLOAD_FOLDER):
@@ -40,12 +40,10 @@ llm = ChatOpenAI(
 )
 
 # Load the CLIP model for image feature extraction
-clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32",from_tf=True)
-clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
 # Function to generate a more appropriate name based on content
-def generate_name_based_on_content(text):
-    prompt = f"Generate a meaningful file name for the following content: {text[:200]}"  # Truncate text to first 200 characters
+def generate_name_based_on_content(text,industry):
+    prompt = f"Generate a meaningful file name for the following content: {text[:200]} based {industry}"  # Truncate text to first 200 characters
     response = llm(prompt)  # Get the model's response
 
     # Extract the generated file name and clean it
@@ -107,7 +105,7 @@ def process_files(files, industry):
                 text = extract_text_from_image(file_path)
 
             # Generate name based on LLM and include timestamp for uniqueness
-            content_name = generate_name_based_on_content(text) if text else 'Untitled'
+            content_name = generate_name_based_on_content(text,industry) if text else 'Untitled'
             directory_name = f"{industry}_{content_name}_{timestamp}"
             new_dir = os.path.join(UPLOAD_FOLDER, directory_name)
             if not os.path.exists(new_dir):