Update app.py

app.py

@@ -2,10 +2,9 @@ import streamlit as st
 import tempfile
 import logging
 from typing import List
-from langchain_community.document_loaders import PyPDFLoader
-from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain_community.vectorstores import FAISS
-from langchain_community.llms import HuggingFacePipeline
+from langchain.document_loaders import PyPDFLoader  # Updated import
+from langchain.embeddings import HuggingFaceEmbeddings  # Updated import
+from langchain.vectorstores import FAISS  # Updated import
 from langchain.chains.summarize import load_summarize_chain
 from langchain.schema import Document
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -35,7 +34,7 @@ def load_llm(model_name):
     """Load and cache the language model."""
     try:
         pipe = pipeline("text2text-generation", model=model_name, max_length=512)
-        return HuggingFacePipeline(pipeline=pipe)
+        return pipe
     except Exception as e:
         logger.error(f"Failed to load LLM: {e}")
         st.error(f"Failed to load the model {model_name}. Please try again.")
@@ -71,46 +70,7 @@ def summarize_report(documents: List[Document], llm) -> str:
     """Summarize the report using the loaded model."""
     try:
         prompt_template = """
-        You are an AI specialized in summarizing comprehensive reports with a focus on funding, finances, and global comparisons. Given the detailed report content below, generate a concise and structured summary using bullet points and emojis. The summary should highlight key funding figures, financial data, budget allocations, comparisons between regions, and notable insights about [FOCUS_REGION]'s role in the global context of [TOPIC].
-
-        Report Content:
-        {text}
-
-        Your summary should follow this structure:
-
-        Summary:
-        💰 [TOPIC] Overview for [FOCUS_REGION]:
-
-        🔴 [FOCUS_REGION]'s Position in Global [TOPIC]:
-        📍 Total investment/funding: [amount]
-        📍 Breakdown of funding sources (e.g., government, private sector)
-        📍 [FOCUS_REGION]'s ranking in global investment
-        📍 Key statistics and figures
-
-        🔴 Financial Impact and Projections:
-        📍 Expected ROI or economic benefits
-        📍 Financial milestones or targets
-        📍 Impact on relevant areas
-
-        🔴 Global Comparison:
-        📍 [List of relevant countries/regions with their financial figures]
-        📍 Comparative analysis of [FOCUS_REGION] vs other major players
-
-        🔴 Budget Analysis:
-        📍 Major budget items
-        📍 Key budget allocations
-        📍 Year-over-year budget changes
-        📍 Comparison to industry benchmarks
-
-        🔴 Funding Strategies:
-        📍 Key funding mechanisms (e.g., grants, loans, public-private partnerships)
-        📍 Innovative financing approaches
-
-        🔴 Progress and Significance:
-        📍 Key achievements or milestones
-        📍 [1-2 concluding points about [FOCUS_REGION]'s role or significance in [TOPIC]]
-
-        Please ensure the summary is concise, informative, and easy to read at a glance. Use precise figures where available and highlight any significant financial trends or insights. The summary should provide a comprehensive overview of both the financial aspects and the broader context of [TOPIC] in [FOCUS_REGION].
+        You are an AI specialized in summarizing comprehensive reports with a focus on funding, finances, and global comparisons. Given the detailed report content below, generate a concise and structured summary using bullet points and emojis...
         """
 
         prompt = PromptTemplate.from_template(prompt_template)
@@ -155,4 +115,4 @@ def main():
                         st.warning("Failed to generate summary. Please try again.")
 
 if __name__ == "__main__":
-    main()
+    main()