Spaces:
Sleeping
Sleeping
| """ | |
| Hugging Face Spaces Gradio app for RAG System | |
| """ | |
| import gradio as gr | |
| import os | |
| import sys | |
| from pathlib import Path | |
| from dotenv import load_dotenv | |
| HERE = Path(__file__).resolve().parent | |
| ROOT = HERE.parent | |
| if str(ROOT) not in sys.path: | |
| sys.path.insert(0, str(ROOT)) | |
| from src.app_hf import answer_question, add_urls | |
| load_dotenv() | |
| # Check if HF_TOKEN is set | |
| hf_token = os.getenv("HF_TOKEN") | |
| def _normalize_history(history): | |
| if not history: | |
| return [] | |
| normalized = [] | |
| for item in history: | |
| if isinstance(item, dict) and item.get("role") in {"user", "assistant"}: | |
| normalized.append(item) | |
| elif isinstance(item, (list, tuple)) and len(item) == 2: | |
| normalized.append({"role": "user", "content": item[0]}) | |
| normalized.append({"role": "assistant", "content": item[1]}) | |
| return normalized | |
| def respond(message, chat_history, urls_text, progress=gr.Progress()): | |
| try: | |
| history = _normalize_history(chat_history) | |
| # Parse URLs from text input | |
| urls = None | |
| if urls_text and urls_text.strip(): | |
| urls = [url.strip() for url in urls_text.split('\n') if url.strip()] | |
| add_urls(urls) | |
| reply = answer_question(message, urls=urls, progress=progress) | |
| history.append({"role": "user", "content": message}) | |
| history.append({"role": "assistant", "content": reply}) | |
| return history, history | |
| except Exception as e: | |
| error_reply = f"❌ Error: {str(e)}" | |
| history = _normalize_history(chat_history) | |
| history.append({"role": "user", "content": message}) | |
| history.append({"role": "assistant", "content": error_reply}) | |
| return history, history | |
| # Build the Gradio interface | |
| with gr.Blocks(title="📚 Documentation RAG Assistant") as demo: | |
| gr.Markdown("# 📚 Documentation RAG Assistant") | |
| gr.Markdown("Ask questions about your documentation with AI-powered retrieval") | |
| gr.Markdown(f"**Status:** {'✅ Ready' if hf_token else '❌ HF_TOKEN not set'}") | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| chatbot = gr.Chatbot(label="Chat") | |
| msg = gr.Textbox( | |
| placeholder="Ask me anything...", | |
| label="Your Question" | |
| ) | |
| state = gr.State([]) | |
| with gr.Column(scale=1): | |
| gr.Markdown("### 📖 Data Sources") | |
| urls_input = gr.Textbox( | |
| placeholder="https://docs.python.org\nhttps://realpython.com", | |
| label="URLs to Search", | |
| lines=5, | |
| info="Paste URLs here (one per line)" | |
| ) | |
| gr.Markdown(""" | |
| **Or** upload PDFs: | |
| - Use the file upload below | |
| - Supports PDF documents | |
| """) | |
| def submit_fn(user_message, history, urls): | |
| history = history or [] | |
| new_history, _ = respond(user_message, history, urls) | |
| return "", new_history, new_history | |
| msg.submit(submit_fn, [msg, state, urls_input], [msg, chatbot, state]) | |
| send_btn = gr.Button("Send", variant="primary") | |
| send_btn.click(submit_fn, [msg, state, urls_input], [msg, chatbot, state]) | |
| if __name__ == "__main__": | |
| demo.queue() | |
| demo.launch() | |