Update app.py

app.py

@@ -57,7 +57,9 @@ class DocumentRAG:
             documents = []
             for uploaded_file in uploaded_files:
                 # Save uploaded file to a temporary location
-                temp_file_path = tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]).name
+                temp_file_path = tempfile.NamedTemporaryFile(
+                    delete=False, suffix=os.path.splitext(uploaded_file.name)[1]
+                ).name
                 with open(temp_file_path, "wb") as temp_file:
                     temp_file.write(uploaded_file.read())
 
@@ -91,7 +93,6 @@ class DocumentRAG:
             # Combine text for later summary generation
             self.document_text = " ".join([doc.page_content for doc in documents])  # Store for later use
 
-
             # Create embeddings and initialize retrieval chain
             embeddings = OpenAIEmbeddings(api_key=self.api_key)
             self.document_store = Chroma.from_documents(
@@ -112,7 +113,6 @@ class DocumentRAG:
         except Exception as e:
             return f"Error processing documents: {str(e)}"
 
-
     def generate_summary(self, text, language):
         """Generate a summary of the provided text focusing on specific sections in the specified language."""
         if not self.api_key:
@@ -123,21 +123,20 @@ class DocumentRAG:
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": f"""
-                    Summarize the following document focusing mainly using the following sections:
-                1. Abstract
-                2. In the introduction section, focus specifically on the portion where the key contributions of the research paper are highlighted.
-                3. Conclusion
-                4. Limitations
-                5. Future Work
-                
-                Ensure the summary is concise, logically ordered, and suitable for {language}. 
-                Provide 5-7 key points for discussion in a structured format."""
-                },
+                    Summarize the following document focusing mainly on these sections:
+                    1. Abstract
+                    2. In the Introduction, specifically focus on the portion where the key contributions of the research paper are highlighted.
+                    3. Conclusion
+                    4. Limitations
+                    5. Future Work
+                    
+                    Ensure the summary is concise, logically ordered, and suitable for {language}. 
+                    Provide 5-7 key points for discussion in a structured format."""},
                     {"role": "user", "content": text[:4000]}
                 ],
                 temperature=0.3
-                )
-                return response.choices[0].message.content
+            )
+            return response.choices[0].message.content
         except Exception as e:
             return f"Error generating summary: {str(e)}"