Backup before author fix

.babelrc

@@ -1,3 +1,3 @@
 {
-    "presets": ["@babel/preset-env", "@babel/preset-react"]
-}
+    "presets": ["@babel/preset-react"]
+} 

.gitignore

@@ -2,4 +2,35 @@
 dist/
 node_modules/
 yarn.lock
-package-lock.json
+package-lock.json
+
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+
+node_modules
+dist
+dist-ssr
+*.local
+.env
+.venv
+venv
+
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+
+__pycache__
+__pycache__/

AI/app.py

@@ -1,103 +1,145 @@
-from typing import Any
-import logging
-from flask import Flask, request, jsonify, current_app, send_from_directory
-from flask_cors import CORS
-from llmproxy import generate
-import sys
-from pymongo import MongoClient
-import bcrypt
+from __future__ import annotations
+
+import sys, json, logging, requests
+from typing import List, Dict, Any, Union
 from dotenv import load_dotenv
+from flask import Flask, request, jsonify, current_app
+from flask_cors import CORS
 import os
-sys.stdout.reconfigure(encoding='utf-8')
-
-app = Flask(__name__, static_folder='public', static_url_path='')
-CORS(app, supports_credentials=True, origins=["https://localhost:5241, https://localhost:4000", "https://localhost:3000"])
-
-@app.route('/api/document', methods=['POST'])
-def respond_document():
-    data = request.get_json()
 
-    document = data.get("attachment", "").strip()
+load_dotenv()
+api_key = os.getenv('api_key')
+end_point = os.getenv('end_point')
 
-    if not document:
-        return jsonify({"response": "Please provide a document"}), 400
+print(f"API Key: {api_key}")
+print(f"Database URL: {end_point}")
 
-    prompt = f""""""
-
-
-    content = ""
+sys.stdout.reconfigure(encoding='utf-8')  
 
+app = Flask(__name__, static_folder='public', static_url_path='')
+CORS(app, supports_credentials=True, origins=[
+    "https://localhost:5241",
+    "https://localhost:4000",
+    "https://localhost:3000"
+])
+
+log = logging.getLogger("pdf_logger")
+
+def _safe_get_json(raw: Union[Dict[str, Any], str], key: str) -> dict:
+    payload = raw.get(key, "") if isinstance(raw, dict) else ""
+    if isinstance(payload, dict):
+        return payload
     try:
-        final_response = generate(
-            model='4o-mini',
-            system=prompt,
-            query=content,
-            temperature=0.0,
-            lastk=0,
-            session_id='rew'
-        )
-
+        cleaned = payload.strip().removeprefix("```json").removesuffix("```").strip()
+        return json.loads(cleaned)
     except Exception as e:
-        current_app.logger.error("Error in generate: %s", str(e))
-        return jsonify({"response": "Internal error during extraction."}), 500
-
-    return jsonify({"final_response": final_response})
-
-@app.route('/api/image', methods=['POST'])
-def respond_image():
-    data = request.get_json()
-
-    document = data.get("attachment", "").strip()
-
-    if not document:
-        return jsonify({"response": "Please provide an image"}), 400
+        current_app.logger.error("Failed JSON parse: %s", e)
+        raise ValueError("Invalid JSON in LLM output")
+
+def _llm_generate(md_text: str, prompt: str) -> dict:
+    raw = generate(
+        model="4o-mini",
+        system=prompt,
+        query=md_text[:8000],   # truncate for cost/speed safety
+        temperature=0.3,
+        lastk=0,
+        session_id="ai-question-gen",
+    )
+
+    return _safe_get_json(raw, "response")
+
+def _cards_from_text(
+    text: str, questionPrompt: str, answerPrompt: str
+) -> List[Dict[str, str]]:
+    questions = _llm_generate(text, questionPrompt).get("questions", [])
+    print(f"questions: {questions}")
+
+    cards = [{"question": q, "answer": ""} for q in questions]
+
+    user_parts = []
+    user_parts.append("### CONTEXT\n" + text[:8000])
+    user_parts.append("### CONTEXT END\n" + "### QUESTIONS\n".join(f"{i+1}. {q}" for i, q in enumerate(questions)))
+    query = "\n\n".join(user_parts)
+
+    answers = _llm_generate(query, answerPrompt).get("answers", [])
+
+    print(f"answers: {answers}")
+
+    for i, a in enumerate(answers):
+        cards[i]["answer"] = a.strip()
+
+    return cards
+
+@app.route("/api/document", methods=["POST"])
+def respond_document():
+    data = request.get_json(force=True, silent=True) or {}
+    raw_text = str(data.get("text", "")).strip()
+    questionPrompt = str(data.get("questionPrompt", ""))
+    answerPrompt = str(data.get("answerPrompt", ""))
 
-    prompt = f""""""
+    # 详细的调试日志
+    current_app.logger.info(f"=== API Request Debug ===")
+    current_app.logger.info(f"Raw data received: {data}")
+    current_app.logger.info(f"Question Prompt: {questionPrompt}")
+    current_app.logger.info(f"answerPrompt: {answerPrompt}")
 
-    content = ""
+    if not raw_text:
+        return jsonify({"response": "Please provide non-empty 'text'."}), 400
 
     try:
-        final_response = generate(
-            model='4o-mini',
-            system=prompt,
-            query=content,
-            temperature=0.0,
-            lastk=0,
-            session_id='rew'
-        )
-
+        cards = _cards_from_text(raw_text, questionPrompt, answerPrompt)
+        current_app.logger.info(f"Final cards generated: {len(cards)}")
+        current_app.logger.info(f"Card questions: {[card.get('question', '') for card in cards]}")
     except Exception as e:
-        current_app.logger.error("Error in generate: %s", str(e))
-        return jsonify({"response": "Internal error during extraction."}), 500
-
-    return jsonify({"final_response": final_response})
-
-
-@app.route('/api/text', methods=['POST'])
-def respond_text():
-    data = request.get_json()
-
-    text = data.get("text", "").strip()
-
-    if not text:
-        return jsonify({"response": "Please provide text"}), 400
-
-    prompt = f""""""
-
-    content = ""
+        current_app.logger.error("Error in /api/document: %s", e)
+        return jsonify({"response": "Failed to generate flash cards."}), 500
+
+    return jsonify({"cards": cards})
+
+def generate(
+        model: str,
+        system: str,
+        query: str,
+        temperature: float | None = None,
+        lastk: int | None = None,
+        session_id: str | None = None,
+        rag_threshold: float | None = 0.5,
+        rag_usage: bool | None = False,
+        rag_k: int | None = 0
+):
+    headers = {
+        'x-api-key': api_key
+    }
+
+    request = {
+        'model': model,
+        'system': system,
+        'query': query,
+        'temperature': temperature,
+        'lastk': lastk,
+        'session_id': session_id,
+        'rag_threshold': rag_threshold,
+        'rag_usage': rag_usage,
+        'rag_k': rag_k
+    }
+
+    print(f"request: {request}")
+
+    msg = None
 
     try:
-        final_response = generate(
-            model='4o-mini',
-            system=prompt,
-            query=content,
-            temperature=0.0,
-            lastk=0,
-            session_id='rew'
-        )
-
-    except Exception as e:
-        current_app.logger.error("Error in generate : %s", str(e))
-        return jsonify({"response": "Internal error during extraction."}), 500
-
-    return jsonify({"final_response": final_response})
+        response = requests.post(end_point, headers=headers, json=request)
+
+        print(f"response: {response}")
+
+        if response.status_code == 200:
+            res = json.loads(response.text)
+            msg = {'response': res['result'], 'rag_context': res['rag_context']}
+        else:
+            msg = f"Error: Received response code {response.status_code}"
+    except requests.exceptions.RequestException as e:
+        msg = f"An error occurred: {e}"
+    return msg
+
+if __name__ == "__main__":
+    app.run(ssl_context=("dev.crt", "dev.key"), host="127.0.0.1", port=5001)