Create app.py

app.py

@@ -0,0 +1,132 @@
+# app.py
+import os
+import re
+import shutil
+from pathlib import Path
+from datetime import datetime
+
+import gradio as gr
+
+ROOT_DIR = Path(__file__).resolve().parent
+PAPERS_DIR = ROOT_DIR / "papers"
+CONFIG_PATH = ROOT_DIR / "config.yaml"
+
+
+def ensure_dirs():
+    PAPERS_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def save_pdf_to_papers(uploaded_file):
+    """
+    Save uploaded PDF into papers/ with a timestamped filename to avoid overwriting.
+    """
+    ensure_dirs()
+
+    if uploaded_file is None:
+        return "❌ No file received. Please upload a PDF."
+
+    # gr.File returns a tempfile path-like object; in recent gradio it's a NamedString / temp path
+    src_path = Path(uploaded_file)
+    if not src_path.exists():
+        return "❌ Uploaded file path not found."
+
+    if src_path.suffix.lower() != ".pdf":
+        return "❌ Only PDF files are supported. Please upload a .pdf file."
+
+    ts = datetime.now().strftime("%Y%m%d_%H%M%S")
+    safe_name = re.sub(r"[^a-zA-Z0-9._-]+", "_", src_path.name)
+    dst_name = f"{ts}_{safe_name}"
+    dst_path = PAPERS_DIR / dst_name
+
+    shutil.copy2(src_path, dst_path)
+    return f"✅ PDF saved to: {dst_path.as_posix()}"
+
+
+def update_gemini_api_key(api_key: str):
+    """
+    Update config.yaml: api_keys -> gemini_api_key: "<api_key>"
+    Keeps other content unchanged as much as possible.
+    """
+    api_key = (api_key or "").strip()
+    if not api_key:
+        return "❌ API key is empty. Please input your Google API Key."
+
+    if not CONFIG_PATH.exists():
+        return f"❌ config.yaml not found at: {CONFIG_PATH.as_posix()}"
+
+    text = CONFIG_PATH.read_text(encoding="utf-8")
+
+    # Strategy:
+    # 1) Try to replace an existing line: gemini_api_key: "..."
+    # 2) If not found, try to insert under api_keys:
+    replaced = False
+
+    # Replace gemini_api_key line with quoted value (handles single/double/no quotes)
+    pattern = r'(^\s*gemini_api_key\s*:\s*)(?:"[^"]*"|\'[^\']*\'|.*)?\s*$'
+    def repl(m):
+        nonlocal replaced
+        replaced = True
+        return f'{m.group(1)}"{api_key}"'
+
+    new_text, n = re.subn(pattern, repl, text, flags=re.MULTILINE)
+    if n > 0 and replaced:
+        CONFIG_PATH.write_text(new_text, encoding="utf-8")
+        return "✅ Gemini API key saved into config.yaml."
+
+    # If gemini_api_key line does not exist, try inserting it inside api_keys block
+    # Find "api_keys:" line and insert next line with indentation (2 spaces)
+    lines = text.splitlines()
+    out = []
+    inserted = False
+    for i, line in enumerate(lines):
+        out.append(line)
+        if not inserted and re.match(r'^\s*api_keys\s*:\s*$', line):
+            # Insert gemini_api_key right after api_keys:
+            out.append(f'  gemini_api_key: "{api_key}"')
+            inserted = True
+
+    if inserted:
+        CONFIG_PATH.write_text("\n".join(out) + ("\n" if text.endswith("\n") else ""), encoding="utf-8")
+        return "✅ Gemini API key inserted into config.yaml under api_keys."
+    else:
+        return '❌ Could not find "api_keys:" section in config.yaml. Please check your config.yaml format.'
+
+
+with gr.Blocks(title="Paper Upload & API Key Setup") as demo:
+    gr.Markdown("# Demo Space")
+    gr.Markdown(
+        "This app lets you upload a paper PDF and save it into the `papers/` folder, "
+        "and input your Google API Key (Gemini) to update `config.yaml`."
+    )
+
+    with gr.Tab("Upload PDF"):
+        gr.Markdown("**Please upload your paper PDF here.**")
+        pdf_input = gr.File(label="Upload paper PDF here", file_types=[".pdf"])
+        pdf_save_btn = gr.Button("Save PDF to papers/")
+        pdf_status = gr.Markdown()
+
+        pdf_save_btn.click(
+            fn=save_pdf_to_papers,
+            inputs=[pdf_input],
+            outputs=[pdf_status],
+        )
+
+    with gr.Tab("API Key"):
+        gr.Markdown("**Please input your Google API Key here.**")
+        api_key_input = gr.Textbox(
+            label="Google API Key (Gemini)",
+            placeholder="Paste your Google API Key here...",
+            type="password",
+        )
+        api_save_btn = gr.Button("Save API Key to config.yaml")
+        api_status = gr.Markdown()
+
+        api_save_btn.click(
+            fn=update_gemini_api_key,
+            inputs=[api_key_input],
+            outputs=[api_status],
+        )
+
+
+if __name__ == "__main__":
+    demo.launch()