Update app.py

app.py

@@ -5,6 +5,10 @@ import os
 import numpy as np
 from sklearn.metrics.pairwise import cosine_similarity
 from docx import Document
+from docx.shared import Pt
+from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
+from docx.oxml.ns import nsdecls
+from docx.oxml import parse_xml
 import io
 import tempfile
 from astroquery.nasa_ads import ADS
@@ -191,10 +195,64 @@ def generate_data_insights(user_input, exoplanet_data, max_tokens=500, temperatu
 
 def export_to_word(response_content):
     doc = Document()
+    
+    # Add a title
     doc.add_heading('AI Generated SCDD', 0)
-    for line in response_content.split('\n'):
-        doc.add_paragraph(line)
+
+    # Split the response by sections (### is used to mark sections)
+    sections = response_content.split('### ')
     
+    for section in sections:
+        if section.strip():
+            # Check for subsections based on section titles
+            if section.startswith('Science Objectives'):
+                doc.add_heading('Science Objectives', level=1)
+                objectives = section.split('\n')[1:]
+                for objective in objectives:
+                    if objective.strip().startswith('1.') or objective.strip().startswith('-'):
+                        doc.add_paragraph(objective.strip(), style='List Number')
+            elif section.startswith('Physical Parameters'):
+                doc.add_heading('Physical Parameters', level=1)
+                params = section.split('\n')[1:]
+                for param in params:
+                    if param.strip():
+                        doc.add_paragraph(param.strip(), style='Normal')
+            elif section.startswith('Observables'):
+                doc.add_heading('Observables', level=1)
+                observables = section.split('\n')[1:]
+                for observable in observables:
+                    if observable.strip():
+                        doc.add_paragraph(observable.strip(), style='Normal')
+            elif section.startswith('Description of Desired Observations'):
+                doc.add_heading('Description of Desired Observations', level=1)
+                observations = section.split('\n')[1:]
+                for obs in observations:
+                    if obs.strip().startswith('1.') or obs.strip().startswith('-'):
+                        doc.add_paragraph(obs.strip(), style='List Number')
+            elif section.startswith('Technical Requirements Table'):
+                doc.add_heading('Technical Requirements Table', level=1)
+                # Extract the table part from the section
+                table_lines = section.split('\n')[2:]
+                # Assuming the table is split by pipes "|", let's convert it to a Word table
+                table_data = [line.split('|')[1:-1] for line in table_lines if '|' in line]
+                if table_data:
+                    # Add the table to the document
+                    table = doc.add_table(rows=len(table_data), cols=len(table_data[0]))
+                    table.style = 'Table Grid'
+                    for i, row in enumerate(table_data):
+                        for j, cell_text in enumerate(row):
+                            cell = table.cell(i, j)
+                            cell.text = cell_text.strip()
+                            # Apply text wrapping for each cell
+                            cell._element.get_or_add_tcPr().append(parse_xml(r'<w:tcW w:w="2500" w:type="pct" ' + nsdecls('w') + '/>'))
+            elif section.startswith('ADS References'):
+                doc.add_heading('ADS References', level=1)
+                references = section.split('\n')[1:]
+                for reference in references:
+                    if reference.strip():
+                        doc.add_paragraph(reference.strip(), style='Normal')
+
+    # Save the document to a temporary file
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
     doc.save(temp_file.name)