Add application file

.cursorignore

@@ -10,3 +10,5 @@ etc
 .vscode
 .env
 .env.local
+har_and_cookies
+output

.gitignore

@@ -21,3 +21,4 @@ rad.json
 #Test results file
 TestResults.xml
 trash
+output

app.py

@@ -9,6 +9,7 @@ app.config.from_object(Config)
 # Register blueprints
 app.register_blueprint(views_bp)
 app.register_blueprint(api_bp)
+app.run(debug=True)
 def create_app():
     app = Flask(__name__)
     app.config.from_object(Config)

routes/api.py

@@ -36,28 +36,55 @@ def extract_chapters(index_content: str) -> List[str]:
                 chapters.append(chapter_title)
     return chapters if chapters else ["Literature Review", "Methodology", "Results and Discussion"]
 
-def generate_automatic_sections(model: str, research_subject: str) -> List[Tuple[str, str]]:
+def generate_automatic_sections(model: str, research_subject: str, chapter_count: str = 'auto', 
+                              word_count: str = 'auto', include_references: bool = False, 
+                              citation_style: str = None) -> List[Tuple[str, str]]:
     """Generate sections automatically based on AI-generated index"""
     try:
-        index_content = model_provider.generate_index_content(model, research_subject)
+        # Generate index with specified parameters
+        index_prompt = f"Create a research paper outline about {research_subject}"
+        if chapter_count != 'auto':
+            index_prompt += f" with exactly {chapter_count} main chapters"
+        if word_count != 'auto':
+            index_prompt += f" targeting approximately {word_count} words"
+        
+        index_content = model_provider.generate_index_content(model, index_prompt)
         chapters = extract_chapters(index_content)
         
+        # If specific chapter count requested, adjust chapters list
+        if chapter_count != 'auto':
+            requested_count = int(chapter_count)
+            if len(chapters) > requested_count:
+                chapters = chapters[:requested_count]
+            elif len(chapters) < requested_count:
+                default_chapters = ["Literature Review", "Methodology", "Results", "Discussion"]
+                chapters.extend(default_chapters[len(chapters):requested_count])
+        
         sections = [
             ("Index", index_content),
-            ("Introduction", f"Write a comprehensive introduction for a research paper about {research_subject}. Include background information, research objectives, and significance of the study.")
+            ("Introduction", f"Write a comprehensive introduction for a research paper about {research_subject}. "
+                           f"{'Target word count: ' + word_count + ' words.' if word_count != 'auto' else ''}")
         ]
         
+        # Add chapter prompts
         for i, chapter in enumerate(chapters, 1):
+            word_guidance = f" Target approximately {int(int(word_count) / (len(chapters) + 2)) } words." if word_count != 'auto' else ''
             sections.append(
                 (f"Chapter {i}: {chapter}", 
                  f"Write a detailed chapter about '{chapter}' for a research paper about {research_subject}. "
-                 f"Provide comprehensive coverage of this aspect, including relevant theories, examples, and analysis.")
+                 f"Provide comprehensive coverage of this aspect, including relevant theories, examples, and analysis.{word_guidance}")
             )
         
         sections.append(
-            ("Conclusion", f"Write a conclusion section for a research paper about {research_subject}. Summarize key findings, discuss implications, and suggest future research directions.")
+            ("Conclusion", f"Write a conclusion section for a research paper about {research_subject}.")
         )
         
+        # Add references section if requested
+        if include_references:
+            sections.append(
+                ("References", f"Generate a references section in {citation_style} format for this research paper about {research_subject}.")
+            )
+        
         return sections
     except Exception as e:
         raise Exception(f"Failed to generate automatic structure: {str(e)}")
@@ -101,6 +128,10 @@ def stream():
     research_subject = request.args.get('subject', '').strip()
     selected_model = request.args.get('model', 'gpt-4o')
     structure_type = request.args.get('structure', 'automatic')
+    chapter_count = request.args.get('chapterCount', 'auto')
+    word_count = request.args.get('wordCount', 'auto')
+    include_references = request.args.get('includeReferences') == 'true'
+    citation_style = request.args.get('citationStyle')
 
     def generate():
         try:
@@ -145,8 +176,14 @@ def stream():
                         "current_step": 1
                     }) + "\n\n"
                     
-                    index_content = model_provider.generate_index_content(selected_model, research_subject)
-                    sections.append(("Index", index_content))
+                    sections = generate_automatic_sections(
+                        selected_model, 
+                        research_subject,
+                        chapter_count,
+                        word_count,
+                        include_references,
+                        citation_style
+                    )
                     
                     steps[1]["status"] = "complete"
                     yield "data: " + json.dumps({
@@ -163,7 +200,7 @@ def stream():
                         "current_step": 2
                     }) + "\n\n"
                     
-                    chapters = extract_chapters(index_content)
+                    chapters = extract_chapters(sections[0][1])
                     
                     # Create sub-steps for each chapter with initial timing info
                     chapter_substeps = [
@@ -450,4 +487,5 @@ def download(filename):
         Config.UPLOAD_FOLDER,
         safe_filename,
         as_attachment=True
-    )
+    )
+    

templates/index.html

@@ -57,6 +57,88 @@
                     </div>
                 </div>
 
+                <!-- New automatic structure options -->
+                <div id="automaticOptions" class="space-y-4">
+                    <div>
+                        <label for="chapterCount" class="block text-sm font-medium text-gray-700 mb-1">
+                            <i class="fas fa-list-ol mr-2"></i>Number of Chapters
+                        </label>
+                        <div class="flex space-x-4 mb-2">
+                            <label class="inline-flex items-center">
+                                <input type="radio" name="chapterCountType" value="auto" checked
+                                    class="h-4 w-4 text-indigo-600 focus:ring-indigo-500">
+                                <span class="ml-2 text-gray-700">Automatic</span>
+                            </label>
+                            <label class="inline-flex items-center">
+                                <input type="radio" name="chapterCountType" value="manual"
+                                    class="h-4 w-4 text-indigo-600 focus:ring-indigo-500">
+                                <span class="ml-2 text-gray-700">Manual</span>
+                            </label>
+                        </div>
+                        <div class="flex space-x-4">
+                            <select id="chapterCount" name="chapterCount" disabled
+                                class="w-full px-4 py-2 border border-gray-300 rounded-md focus:ring-indigo-500 focus:border-indigo-500">
+                                <option value="3">3 Chapters</option>
+                                <option value="4">4 Chapters</option>
+                                <option value="5">5 Chapters</option>
+                                <option value="6">6 Chapters</option>
+                            </select>
+                            <input type="number" id="customChapterCount" name="customChapterCount"
+                                class="hidden w-full px-4 py-2 border border-gray-300 rounded-md focus:ring-indigo-500 focus:border-indigo-500"
+                                min="1" max="10" placeholder="Enter number (1-10)">
+                        </div>
+                    </div>
+
+                    <div>
+                        <label for="wordCount" class="block text-sm font-medium text-gray-700 mb-1">
+                            <i class="fas fa-font mr-2"></i>Target Word Count
+                        </label>
+                        <div class="flex space-x-4 mb-2">
+                            <label class="inline-flex items-center">
+                                <input type="radio" name="wordCountType" value="auto" checked
+                                    class="h-4 w-4 text-indigo-600 focus:ring-indigo-500">
+                                <span class="ml-2 text-gray-700">Automatic</span>
+                            </label>
+                            <label class="inline-flex items-center">
+                                <input type="radio" name="wordCountType" value="manual"
+                                    class="h-4 w-4 text-indigo-600 focus:ring-indigo-500">
+                                <span class="ml-2 text-gray-700">Manual</span>
+                            </label>
+                        </div>
+                        <div class="flex space-x-4">
+                            <select id="wordCount" name="wordCount" disabled
+                                class="w-full px-4 py-2 border border-gray-300 rounded-md focus:ring-indigo-500 focus:border-indigo-500">
+                                <option value="2000">2,000 words</option>
+                                <option value="3000">3,000 words</option>
+                                <option value="5000">5,000 words</option>
+                                <option value="8000">8,000 words</option>
+                                <option value="10000">10,000 words</option>
+                            </select>
+                            <input type="number" id="customWordCount" name="customWordCount"
+                                class="hidden w-full px-4 py-2 border border-gray-300 rounded-md focus:ring-indigo-500 focus:border-indigo-500"
+                                min="500" max="20000" step="100" placeholder="Enter words (500-20000)">
+                        </div>
+                    </div>
+
+                    <div>
+                        <label class="block text-sm font-medium text-gray-700 mb-2">
+                            <i class="fas fa-quote-right mr-2"></i>References
+                        </label>
+                        <div class="flex items-center mb-2">
+                            <input type="checkbox" id="includeReferences" name="includeReferences"
+                                class="h-4 w-4 text-indigo-600 focus:ring-indigo-500">
+                            <label for="includeReferences" class="ml-2 text-sm text-gray-700">Include References Section</label>
+                        </div>
+                        <select id="citationStyle" name="citationStyle" disabled
+                            class="w-full px-4 py-2 border border-gray-300 rounded-md focus:ring-indigo-500 focus:border-indigo-500">
+                            <option value="APA">APA</option>
+                            <option value="MLA">MLA</option>
+                            <option value="Chicago">Chicago</option>
+                            <option value="Harvard">Harvard</option>
+                        </select>
+                    </div>
+                </div>
+
                 <div class="pt-4">
                     <button type="submit" id="generateBtn"
                         class="w-full flex justify-center items-center py-3 px-4 border border-transparent rounded-md shadow-sm text-sm font-medium text-white bg-indigo-600 hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-indigo-500 disabled:opacity-50 disabled:cursor-not-allowed">
@@ -359,6 +441,97 @@
                 generateBtn.disabled = false;
             };
         });
+
+        document.addEventListener('DOMContentLoaded', function() {
+            // Get elements
+            const structureRadios = document.querySelectorAll('input[name="structure"]');
+            const automaticOptions = document.getElementById('automaticOptions');
+            const includeReferences = document.getElementById('includeReferences');
+            const citationStyle = document.getElementById('citationStyle');
+            const citationStyleContainer = citationStyle.parentElement;
+
+            // Handle structure radio changes
+            structureRadios.forEach(radio => {
+                radio.addEventListener('change', function() {
+                    automaticOptions.style.display = 
+                        this.value === 'automatic' ? 'block' : 'none';
+                });
+            });
+
+            // Initialize automatic options visibility
+            automaticOptions.style.display = 
+                document.querySelector('input[name="structure"]:checked').value === 'automatic' 
+                    ? 'block' 
+                    : 'none';
+
+            // Handle references checkbox
+            includeReferences.addEventListener('change', function() {
+                citationStyleContainer.style.display = this.checked ? 'block' : 'none';
+                citationStyle.disabled = !this.checked;
+            });
+            // Initialize citation style visibility
+            citationStyleContainer.style.display = includeReferences.checked ? 'block' : 'none';
+
+            // Chapter count controls
+            const chapterCountTypeRadios = document.querySelectorAll('input[name="chapterCountType"]');
+            const chapterCountSelect = document.getElementById('chapterCount');
+            const customChapterCount = document.getElementById('customChapterCount');
+            const chapterCountContainer = chapterCountSelect.parentElement;
+
+            chapterCountTypeRadios.forEach(radio => {
+                radio.addEventListener('change', function() {
+                    const isManual = this.value === 'manual';
+                    chapterCountContainer.style.display = isManual ? 'flex' : 'none';
+                    chapterCountSelect.disabled = !isManual;
+                    customChapterCount.classList.toggle('hidden', true);
+                    chapterCountSelect.classList.toggle('hidden', false);
+                });
+            });
+            // Initialize chapter count visibility
+            chapterCountContainer.style.display = 
+                document.querySelector('input[name="chapterCountType"]:checked').value === 'manual' 
+                    ? 'flex' 
+                    : 'none';
+
+            chapterCountSelect.addEventListener('change', function() {
+                if (this.value === 'custom') {
+                    this.classList.toggle('hidden', true);
+                    customChapterCount.classList.toggle('hidden', false);
+                }
+            });
+
+            // Word count controls
+            const wordCountTypeRadios = document.querySelectorAll('input[name="wordCountType"]');
+            const wordCountSelect = document.getElementById('wordCount');
+            const customWordCount = document.getElementById('customWordCount');
+            const wordCountContainer = wordCountSelect.parentElement;
+
+            wordCountTypeRadios.forEach(radio => {
+                radio.addEventListener('change', function() {
+                    const isManual = this.value === 'manual';
+                    wordCountContainer.style.display = isManual ? 'flex' : 'none';
+                    wordCountSelect.disabled = !isManual;
+                    customWordCount.classList.toggle('hidden', true);
+                    wordCountSelect.classList.toggle('hidden', false);
+                });
+            });
+            // Initialize word count visibility
+            wordCountContainer.style.display = 
+                document.querySelector('input[name="wordCountType"]:checked').value === 'manual' 
+                    ? 'flex' 
+                    : 'none';
+
+            wordCountSelect.addEventListener('change', function() {
+                if (this.value === 'custom') {
+                    this.classList.toggle('hidden', true);
+                    customWordCount.classList.toggle('hidden', false);
+                }
+            });
+
+            // Add custom option to selects
+            chapterCountSelect.innerHTML += '<option value="custom">Custom...</option>';
+            wordCountSelect.innerHTML += '<option value="custom">Custom...</option>';
+        });
     </script>
 </body>
 </html>