Merge pull request #42 from GYFX35/add-document-specialist-role-6497180777360688900

Add Document Specialist role

Author: GYFX35

app.py

@@ -922,6 +922,17 @@ def cartography_assistance_endpoint():
     return jsonify({"status": "success", "message": message})
 
 
+@app.route('/api/v1/assistance/document', methods=['POST'])
+@require_api_key
+def document_assistance_endpoint():
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": _("Prompt is required")}), 400
+    message = google_ai.provide_document_assistance(prompt)
+    return jsonify({"status": "success", "message": message})
+
+
 @app.route('/api/register', methods=['POST'])
 def register():
     data = request.get_json()

frontend/static/js/script.js

@@ -249,6 +249,44 @@ document.addEventListener('DOMContentLoaded', () => {
         });
     }
 
+    // --- Document Specialist ---
+    const documentSpecialistBtn = document.getElementById('document-specialist-btn');
+    if (documentSpecialistBtn) {
+        documentSpecialistBtn.addEventListener('click', async () => {
+            const input = document.getElementById('document-specialist-input');
+            const responseContainer = document.getElementById('document-specialist-response');
+            const apiKey = prompt("Please enter your API key to use the Document Specialist:");
+
+            if (!apiKey) {
+                responseContainer.textContent = 'API key is required.';
+                return;
+            }
+
+            try {
+                const response = await fetch('/api/v1/assistance/document', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        'X-API-Key': apiKey
+                    },
+                    body: JSON.stringify({
+                        prompt: input.value
+                    })
+                });
+
+                if (!response.ok) {
+                    const error = await response.json();
+                    throw new Error(error.error || 'Failed to get a response from the document specialist');
+                }
+
+                const result = await response.json();
+                responseContainer.textContent = result.message;
+            } catch (error) {
+                responseContainer.textContent = `Error: ${error.message}`;
+            }
+        });
+    }
+
     // --- Promote Startup ---
     const promoteStartupBtn = document.getElementById('promote-startup-btn');
     if (promoteStartupBtn) {

frontend/templates/index.html

@@ -77,6 +77,22 @@ <h3>{{ _('Music Instrumentalist Player') }}</h3>
             </div>
         </section>
 
+        <!-- Document Specialist Section -->
+        <section id="document-specialist" class="services">
+            <h2>{{ _('Document Specialist') }}</h2>
+            <div class="service-cards">
+                <div class="card">
+                    <h3>{{ _('Write, scan, and build documents') }}</h3>
+                    <p>{{ _('Enter your request for ebooks, articles, PDFs, or DOCs below.') }}</p>
+                    <textarea id="document-specialist-input" placeholder="{{ _('Your document request') }}" class="styled-textarea" rows="4"></textarea>
+                    <button id="document-specialist-btn" class="btn">{{ _('Submit') }}</button>
+                    <div class="response-container" id="document-specialist-response-container">
+                        <pre id="document-specialist-response"></pre>
+                    </div>
+                </div>
+            </div>
+        </section>
+
         <!-- Geometry Assistant Section -->
         <section id="geometry-assistant" class="services">
             <h2>{{ _('Geometry Assistant') }}</h2>

google_ai.py

@@ -531,3 +531,33 @@ def provide_cartography_assistance(prompt: str) -> str:
     except Exception as e:
         print(f"Error providing cartography assistance with Vertex AI: {e}")
         return f"Error: {e}"
+
+
+def provide_document_assistance(prompt: str) -> str:
+    """
+    Provides assistance with writing, scanning, and building ebooks, articles, PDFs, and DOCs using Vertex AI.
+    """
+    model = GenerativeModel("gemini-1.5-flash")
+
+    generation_prompt = f"""
+    You are an expert document specialist. Your task is to provide assistance with writing, scanning, and building ebooks, articles, PDFs, DOCs, and other file formats.
+    Your expertise includes:
+    - Writing: Generating high-quality content for ebooks, articles, and reports.
+    - Scanning: Summarizing, extracting key information, and analyzing document content.
+    - Building: Advising on document structure, formatting, and conversion between different formats.
+
+    User Request:
+    ---
+    {prompt}
+    ---
+
+    Provide a professional, detailed, and helpful response based on the user's request.
+    """
+
+    try:
+        response = model.generate_content(generation_prompt)
+        return response.text.strip()
+
+    except Exception as e:
+        print(f"Error providing document assistance with Vertex AI: {e}")
+        return f"Error: {e}"