Update functions.py

functions.py

@@ -15,9 +15,10 @@ highlights_options = {
     "highlights_per_url": 1,  # Get the best highlight for each URL
 }
 
+
 def get_report_fields(report_type):
     """
-    Return a list of fields (field_key, field_label) based on the report type.
+    Return a list of fields (tuples of field_key, field_label) based on the report type.
     """
     common_fields = [
         ("research_topic", "Research Topic"),
@@ -29,23 +30,51 @@ def get_report_fields(report_type):
     ]
 
     if report_type == "PhD Proposal":
-        return common_fields + [
-            ("justification", "Justification for Methodology"),
-            ("literature_gap", "Literature Gap")
-        ]
+        return common_fields + [("justification", "Justification for Methodology"),
+                                ("literature_gap", "Literature Gap")]
+
     elif report_type == "Research Paper":
-        return common_fields + [
-            ("introduction", "Introduction"),
-            ("conclusion", "Conclusion")
-        ]
+        return common_fields + [("introduction", "Introduction"), 
+                                ("conclusion", "Conclusion")]
+
     elif report_type == "Thesis":
-        return common_fields + [
-            ("abstract", "Abstract"),
-            ("limitations", "Limitations")
-        ]
+        return common_fields + [("abstract", "Abstract"),
+                                ("limitations", "Limitations")]
 
     return common_fields
+
+def get_report_sections(report_type):
+    """
+    Return a dictionary mapping section names to corresponding functions based on report type.
+    """
+    report_sections = {
+        "PhD Proposal": [
+            ("Executive Summary", "generate_executive_summary"),
+            ("Research Objectives", "generate_research_objectives"),
+            ("Research Methodology", "generate_methodology_section"),
+            ("Research Timeline", "generate_research_timeline")
+        ],
+        "Research Paper": [
+            ("Introduction", "generate_proposal_introduction"),
+            ("Literature Review", "generate_literature_review_outline"),
+            ("Research Methodology", "generate_methodology_section"),
+            ("Conclusion", "generate_contribution_statement")
+        ],
+        "Thesis": [
+            ("Abstract", "generate_proposal_introduction"),
+            ("Research Objectives", "generate_research_objectives"),
+            ("Methodology", "generate_methodology_section"),
+            ("Limitations", "generate_limitations_section")
+        ]
+    }
+    return report_sections.get(report_type, [])
+
+# Define the actual generation functions, like:
+def generate_executive_summary(data):
+    return f"Executive Summary for {data['research_topic']}"
     
+# Add other report generation functions similar to this
+
 def call_llm(prompt):
     """
     Call the Exa and Groq APIs to generate content using the provided prompt.