Update functions.py

functions.py

@@ -15,6 +15,37 @@ 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.
+    """
+    common_fields = [
+        ("research_topic", "Research Topic"),
+        ("research_question", "Research Question"),
+        ("objectives", "Research Objectives (SMART)"),
+        ("methodology", "Research Methodology"),
+        ("timeline", "Research Timeline"),
+        ("email", "Contact Email"),
+    ]
+
+    if report_type == "PhD Proposal":
+        return common_fields + [
+            ("justification", "Justification for Methodology"),
+            ("literature_gap", "Literature Gap")
+        ]
+    elif report_type == "Research Paper":
+        return common_fields + [
+            ("introduction", "Introduction"),
+            ("conclusion", "Conclusion")
+        ]
+    elif report_type == "Thesis":
+        return common_fields + [
+            ("abstract", "Abstract"),
+            ("limitations", "Limitations")
+        ]
+
+    return common_fields
+    
 def call_llm(prompt):
     """
     Call the Exa and Groq APIs to generate content using the provided prompt.