Update pages/Report_Writer.py

pages/Report_Writer.py

@@ -14,9 +14,56 @@ from llama_index.core.node_parser import TokenTextSplitter
 
 # Configure Google Gemini
 Settings.embed_model = FastEmbedEmbedding(model_name="BAAI/bge-small-en-v1.5")
-Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
+Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.1, model_name="models/gemini-pro")
 llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.1, model_name="models/gemini-pro")
 
+DEFAULT_REPORT_FORMAT = """
+    Title Page
+
+        Includes the report title, author's name, and date.
+
+    Abstract
+
+        A concise summary of the report, covering the background, objectives, methodology, key findings, and conclusions.
+
+    Table of Contents
+
+        Lists sections and subsections with corresponding page numbers for easy navigation.
+
+    Introduction
+
+        Provides background information, defines the scope of the report, and states the objectives.
+
+    Literature Review
+
+        Reviews relevant literature and previous research related to the report topic.
+
+    Methodology/Approach
+
+        Details the methods used to gather data or conduct experiments, including design and analytical techniques.
+
+    Results and Discussion
+
+        Presents findings in a clear format, often using tables, figures, and charts, followed by a discussion interpreting these results.
+
+    Conclusions
+
+        Summarizes the main findings and their implications, often linking back to the report's objectives.
+
+    Recommendations
+
+        Suggests actions based on the findings, highlighting potential future work or improvements.
+
+    References
+
+        Lists all sources cited in the report, adhering to a specific referencing style.
+
+    Appendices
+
+        Contains supplementary material that supports the main text, such as raw data, detailed calculations, or additional figures.
+
+"""
+
 # Load and index the input data
 def load_data(document_text):
     # Use a text splitter to break the document into smaller chunks