Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import time | |
| from utils import ( | |
| get_question, | |
| get_answer, | |
| evaluate_expertise, | |
| get_crypto_data, | |
| ) | |
| def run_chat(crypto, data_span): | |
| # Validate inputs | |
| valid_cryptos = ["btc", "eth", "virtual", "bnb"] | |
| valid_spans = ["24h", "7d", "14d", "30d", "60d"] | |
| if crypto not in valid_cryptos or data_span not in valid_spans: | |
| yield [("🤖", "❌ Invalid crypto or data span selection")], [] | |
| return | |
| crypto_data = get_crypto_data(crypto, data_span) | |
| chat_history = [] | |
| previous_qa = "" | |
| qa_history = [] | |
| for _ in range(5): | |
| # --- QUESTION --- | |
| question = get_question(previous_qa) | |
| # append as a right-only message | |
| chat_history.append(("", question.content)) | |
| yield chat_history, [] # show question | |
| time.sleep(1) | |
| # --- ANSWER --- | |
| answer = get_answer(question.content, previous_qa, crypto_data) | |
| # append as a left-only message | |
| chat_history.append((answer.content, "")) | |
| yield chat_history, [] # show answer | |
| # track for final evaluation | |
| qa_history.append({"question": question.content, "answer": answer.content}) | |
| previous_qa += f"\nQ: {question.content}\nA: {answer.content}\n" | |
| time.sleep(1) | |
| # --- FINAL EVALUATION --- | |
| evaluation = evaluate_expertise(qa_history) | |
| chat_history.append(("📊 Evaluation:", evaluation.content)) | |
| yield chat_history, chat_history | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## 🧠 TranchVest AI - Crypto Behavior Analysis") | |
| with gr.Row(): | |
| crypto_dropdown = gr.Dropdown( | |
| ["btc", "eth", "virtual", "bnb"], label="Select Crypto" | |
| ) | |
| span_dropdown = gr.Dropdown( | |
| ["24h", "7d", "14d", "30d", "60d"], label="Select Data Span" | |
| ) | |
| start_btn = gr.Button("Start Chat") | |
| chatbot = gr.Chatbot(height=600) | |
| state = gr.State([]) | |
| start_btn.click( | |
| run_chat, | |
| [crypto_dropdown, span_dropdown], | |
| [chatbot, state], | |
| ) | |
| demo.launch() | |