updated for resume translation

app.py

@@ -16,7 +16,7 @@ dwani.api_base = os.getenv("DWANI_API_BASE_URL")
 logger.debug("DWANI_API_KEY: %s", "Set" if dwani.api_key else "Not set")
 logger.debug("DWANI_API_BASE_URL: %s", dwani.api_base)
 
-# Language options for dropdowns
+# Language options for dropdowns (display name and code)
 language_options = [
     ("English", "eng_Latn"),
     ("Kannada", "kan_Knda"),
@@ -60,18 +60,35 @@ def parse_page_numbers(pages_str):
 
 def results_to_markdown(results):
     """
-    Convert the results dictionary into a Markdown formatted string.
+    Convert the results dictionary into a Markdown formatted string,
+    formatting the translated response to preserve structure using <pre> tags.
     """
     md_lines = []
     for page, content in results.items():
-        md_lines.append(f"## {page}")
+        md_lines.append(f"## {page}\n")
         if "error" in content:
-            md_lines.append(f"**Error:** {content['error']}")
+            md_lines.append(f"**Error:** {content['error']}\n")
         else:
-            md_lines.append(f"**Original Text:**\n\n``````")
-            md_lines.append(f"**Response:**\n\n{content.get('Response', '')}")
-            md_lines.append(f"**Processed Page:** {content.get('Processed Page', '')}")
-            md_lines.append(f"**Translated Response:**\n\n{content.get('Translated Response', '')}")
+            md_lines.append("**Original Text:**\n\n```")
+            md_lines.append(content.get('Original Text', '') + "\n")
+            md_lines.append("```\n")
+
+            response_text = content.get('Response', '')
+            if response_text:
+                md_lines.append("Response:\n\n" + response_text + "\n")
+
+            md_lines.append("**Processed Page:** " + str(content.get('Processed Page', '')) + "\n")
+
+            translated = content.get('Translated Response', '')
+
+            # Normalize newlines
+            translated = translated.replace('\r\n', '\n').replace('\r', '\n')
+
+            # Use <pre> tags to preserve formatting exactly
+            md_lines.append("**Translated Response:**\n\n<pre>")
+            md_lines.append(translated)
+            md_lines.append("</pre>")
+
         md_lines.append("\n---\n")
     return "\n".join(md_lines)
 
@@ -108,24 +125,34 @@ def process_pdf(pdf_file, pages_str, prompt, src_lang, tgt_lang):
     logger.debug("Calling API with file: %s, pages: %s, prompt: %s, src_lang: %s, tgt_lang: %s",
                  file_path, pages, prompt, src_lang_code, tgt_lang_code)
 
-    system_prompt = "Do not return any asterisk"
-
     results = {}
     for page_number in pages:
         try:
-            result = dwani.Documents.run_doc_query(
+            result = dwani.Documents.run_extract(
                 file_path=file_path,
-                prompt=f"{prompt} {system_prompt}",
                 page_number=page_number,
                 src_lang=src_lang_code,
                 tgt_lang=tgt_lang_code
             )
             logger.debug("API response for page %d: %s", page_number, result)
+
+            # New response format: result contains 'pages' list
+            page_data = None
+            for p in result.get('pages', []):
+                if p.get('processed_page') == page_number:
+                    page_data = p
+                    break
+
+            if page_data is None:
+                results[f"Page {page_number}"] = {"error": "No data returned for this page"}
+                continue
+
             results[f"Page {page_number}"] = {
-                "Original Text": result.get("original_text", "N/A"),
-                "Response": result.get("response", "N/A"),
-                "Processed Page": result.get("processed_page", "N/A"),
-                "Translated Response": result.get("translated_response", "N/A")
+                "Processed Page": page_data.get("processed_page", "N/A"),
+                "Original Text": page_data.get("page_content", "N/A"),
+                "Translated Response": page_data.get("translated_content", "N/A"),
+                # The old 'Response' key is not in new data; set empty string
+                "Response": ""
             }
         except dwani.exceptions.DhwaniAPIError as e:
             logger.error("Dhwani API error on page %d: %s", page_number, str(e))