Spaces:
Running on Zero
Running on Zero
| import logging | |
| import time | |
| import gradio as gr | |
| import spaces | |
| from support_rag import ask_support | |
| logging.basicConfig(level=logging.INFO) | |
| logger = logging.getLogger("experiment7.gradio") | |
| # Hugging Face ZeroGPU Spaces expect at least one GPU-decorated function. | |
| # This no-op is not called by the RAG pipeline; inference is handled by Groq. | |
| def zerogpu_compatibility_check(): | |
| return "ZeroGPU Space is configured." | |
| def answer_question(question): | |
| question = (question or "").strip() | |
| if not question: | |
| return ( | |
| "Please enter a question.", | |
| "-", | |
| "[]", | |
| "0", | |
| "0 ms", | |
| ) | |
| start = time.perf_counter() | |
| try: | |
| result = ask_support(question) | |
| elapsed_ms = round((time.perf_counter() - start) * 1000, 2) | |
| sources = result.get("sources", []) | |
| return ( | |
| result.get("answer", "Sorry, I could not generate an answer."), | |
| result.get("intent", "general_support"), | |
| ", ".join(sources) if sources else "No relevant FAQ source found", | |
| str(result.get("retry_count", 0)), | |
| f"{elapsed_ms} ms", | |
| ) | |
| except Exception: | |
| logger.exception("Customer-support request failed.") | |
| return ( | |
| "The support service encountered an error. Please try again later. " | |
| "Check the Space logs for technical details.", | |
| "error", | |
| "[]", | |
| "0", | |
| "Error", | |
| ) | |
| with gr.Blocks(title="Customer Support RAG") as demo: | |
| gr.Markdown( | |
| "# Customer Support RAG Assistant\n" | |
| "Ask questions about order tracking, returns, refunds, account access, " | |
| "billing, or technical support. Answers are grounded in the FAQ." | |
| ) | |
| question = gr.Textbox( | |
| label="Customer question", | |
| placeholder="How can I request a refund?", | |
| lines=2, | |
| max_lines=4, | |
| ) | |
| submit = gr.Button("Ask", variant="primary") | |
| answer = gr.Textbox(label="Answer", lines=4) | |
| with gr.Row(): | |
| intent = gr.Textbox(label="Detected intent") | |
| source = gr.Textbox(label="Retrieved source") | |
| with gr.Row(): | |
| retries = gr.Textbox(label="Retry count") | |
| response_time = gr.Textbox(label="Response time") | |
| submit.click( | |
| fn=answer_question, | |
| inputs=question, | |
| outputs=[answer, intent, source, retries, response_time], | |
| ) | |
| question.submit( | |
| fn=answer_question, | |
| inputs=question, | |
| outputs=[answer, intent, source, retries, response_time], | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |