| import gradio as gr |
| from llama_index.readers.file import PagedCSVReader |
| from llama_index.core.indices import VectorStoreIndex |
| import openai |
| import os |
|
|
| openai.api_key = os.getenv('OPENAI_API_KEY') |
|
|
| def load_data(): |
| try: |
| loader = PagedCSVReader() |
| documents = loader.load_data('aitalents.csv') |
| index = VectorStoreIndex.from_documents(documents) |
| query_engine = index.as_query_engine() |
| return query_engine |
| except Exception as e: |
| print(f"Error loading data or creating index: {e}") |
| return None |
|
|
| query_engine = load_data() |
|
|
| def chat(message, history): |
| if query_engine is None: |
| return "An error occurred while loading data. Please try again later." |
|
|
| try: |
| response = query_engine.query(message) |
| return str(response) |
| except Exception as e: |
| print(f"Error generating response: {e}") |
| return "I'm still learning how to answer that question. Please try asking something else." |
|
|
| |
| interface = gr.ChatInterface(fn=chat, title="AI Talent Matchmaker") |
|
|
| |
| interface.launch() |
