Rename app.py to streamlit_app.py

app.py → streamlit_app.py

@@ -27,8 +27,61 @@ def load_rewriter(model_name):
     return pipeline("text2text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
 
 HEADERS = [
-    "Abstract", "Introduction", "Background", "Related Work", "Literature Review", "Methodology", "Results", 
-    "Discussion", "Conclusion", "References", "Appendix", "Acknowledgements"
+    # Standard sections
+    "Abstract", "Introduction", "Background",
+    "Related Work", "Literature Review", "Theoretical Framework",
+    "Methodology", "Research Methodology", "Materials and Methods",
+    "Experimental Setup", "Experimental Design", "Research Design",
+    "Implementation", "System Architecture", "Algorithm",
+    "Data Collection", "Data Analysis", "Statistical Analysis",
+    "Results", "Findings", "Experimental Results", "Case Study",
+    "Discussion", "Analysis", "Interpretation", "Evaluation",
+    "Conclusion", "Conclusions", "Summary", "Future Work",
+    "Recommendations", "Implications", "Limitations",
+    "References", "Bibliography", "Acknowledgements",
+
+    # Subject-specific sections
+    # STEM
+    "Hypothesis", "Procedure", "Observations",
+    "Calculations", "Derivation", "Proof",
+    "Simulation", "Modeling", "Validation",
+    "Performance Metrics", "Benchmarking",
+
+    # Medical/Health Sciences
+    "Clinical Trial", "Patient Characteristics",
+    "Intervention", "Outcome Measures",
+    "Adverse Effects", "Ethical Considerations",
+
+    # Social Sciences
+    "Research Questions", "Conceptual Framework",
+    "Participant Demographics", "Survey Instrument",
+    "Qualitative Analysis", "Quantitative Analysis",
+    "Themes", "Coding Scheme",
+
+    # Business/Economics
+    "Market Analysis", "Financial Projections",
+    "SWOT Analysis", "Case Analysis",
+    "Return on Investment", "Cost-Benefit Analysis",
+
+    # Humanities
+    "Historical Context", "Textual Analysis",
+    "Critical Review", "Comparative Analysis",
+    "Theoretical Perspective",
+
+    # Engineering/CS
+    "System Requirements", "Technical Specifications",
+    "Pseudocode", "Flowchart", "UML Diagram",
+    "User Interface", "API Documentation",
+    "Error Analysis", "Optimization",
+
+    # Appendices
+    "Appendix A", "Appendix B", "Supplementary Materials",
+    "Additional Results", "Raw Data",
+
+    # Conference/Journal Specific
+    "Key Takeaways", "Author Contributions",
+    "Conflict of Interest", "Data Availability",
+    "Supplementary Information"
 ]
 header_pattern = re.compile(rf"\n*(?:\d{{0,2}}[\.\)]?\s*)?({'|'.join(HEADERS)})\s*\n", re.IGNORECASE)