Update summarizer.py

summarizer.py

@@ -4,27 +4,53 @@ from langchain_cohere.llms import Cohere
 from langchain.chains.summarize import load_summarize_chain
 from pathlib import Path
 
+
 def summarize_files(method, files):
     # Initialize the LLM
     llm = Cohere(temperature=0)
-    summaries = []
-    # Load and read each file
-    for file in files:
+
+    ext = Path(file).suffix.lower()
+    if ext == '.pdf':
+        loader = PyPDFLoader(file)
+    elif ext == '.docx':
+        loader = Docx2txtLoader(file)
+    elif ext == '.pptx':
+        loader = UnstructuredPowerPointLoader(file)
+    else:
+        raise ValueError(f"Unsupported file extension: {ext}")
+    
+    docs = loader.load_and_split()
+    # Initialize a summarization chain with the specified method
+    summarization_chain = load_summarize_chain(llm=llm, chain_type=method)
+    summary = summarization_chain.run(docs)
+    
+    return summary
+
+
+
+
+
+# def summarize_files(method, files):
+#     # Initialize the LLM
+#     llm = Cohere(temperature=0)
+#     summaries = []
+#     # Load and read each file
+#     for file in files:
         
-        ext = Path(file).suffix.lower()
-        if ext == '.pdf':
-            loader = PyPDFLoader(file)
-        elif ext == '.docx':
-            loader = Docx2txtLoader(file)
-        elif ext == '.pptx':
-            loader = UnstructuredPowerPointLoader(file)
-        else:
-            raise ValueError(f"Unsupported file extension: {ext}")
+#         ext = Path(file).suffix.lower()
+#         if ext == '.pdf':
+#             loader = PyPDFLoader(file)
+#         elif ext == '.docx':
+#             loader = Docx2txtLoader(file)
+#         elif ext == '.pptx':
+#             loader = UnstructuredPowerPointLoader(file)
+#         else:
+#             raise ValueError(f"Unsupported file extension: {ext}")
         
-        docs = loader.load_and_split()
-        # Initialize a summarization chain with the specified method
-        summarization_chain = load_summarize_chain(llm=llm, chain_type=method)
-        summary = summarization_chain.run(docs)
-        summaries.append(summary)    
+#         docs = loader.load_and_split()
+#         # Initialize a summarization chain with the specified method
+#         summarization_chain = load_summarize_chain(llm=llm, chain_type=method)
+#         summary = summarization_chain.run(docs)
+#         summaries.append(summary)    
     
-    return summaries
+#     return summaries