๐
Research & Search
Web search, Wikipedia, academic papers, arXiv research
๐ป
Code Execution
Python, Bash, SQL, C, Java with real-time results
๐
Data Analysis
CSV, Excel, visualizations, statistical analysis
๐ผ๏ธ
Image Processing
Analysis, OCR, transformations, generation
---
"""
)
# Chat interface with enhanced styling
with gr.Row(elem_classes="main-content"):
with gr.Column(scale=1):
chatbot_interface = gr.Chatbot(
label="๐ฌ Conversation",
height=600,
show_label=True,
container=True,
bubble_full_width=False,
elem_classes="chatbot"
)
# File upload section with enhanced styling
with gr.Row(elem_classes="main-content"):
with gr.Column():
file_upload = gr.File(
label="๐ Upload Files - Drag & drop or click to upload images, documents, CSV, Excel files, etc.",
file_count="multiple",
file_types=[
".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp", # Images
".txt", ".md", ".py", ".js", ".html", ".css", ".json", ".xml", # Text files
".csv", ".xlsx", ".xls", # Data files
".pdf", ".doc", ".docx" # Documents
],
height=120,
elem_classes="file-upload"
)
# Input and buttons with enhanced styling
with gr.Row(elem_classes="main-content"):
with gr.Column(scale=8):
question_input = gr.Textbox(
label="๐ญ Ask a question",
placeholder="Type your question here or upload files above... (e.g., 'What is the capital of France?', 'Analyze this image', 'Summarize this document')",
lines=3,
max_lines=5,
elem_classes="textbox"
)
with gr.Column(scale=2, min_width=120):
submit_btn = gr.Button("๐ Send", variant="primary", size="lg", elem_classes="btn btn-primary")
with gr.Row(elem_classes="main-content"):
with gr.Column(scale=1):
clear_btn = gr.Button("๐งน Clear History", variant="secondary", elem_classes="btn btn-secondary")
with gr.Column(scale=1):
clear_files_btn = gr.Button("๐๏ธ Clear Files", variant="secondary", elem_classes="btn btn-secondary")
with gr.Column(scale=1):
export_btn = gr.Button("๐พ Export Chat", variant="secondary", elem_classes="btn btn-secondary")
# Hidden download component for chat export
download_file = gr.File(visible=False)
with gr.Row(elem_classes="main-content"):
with gr.Column():
gr.Examples(
examples=[
"What is the current population of Tokyo?",
"Calculate the square root of 144",
"Write a Python function to sort a list",
"What are the latest developments in AI?",
"Explain quantum computing in simple terms",
],
inputs=question_input,
label="๐ General Questions"
)
with gr.Column():
gr.Examples(
examples=[
"Search for recent papers on machine learning",
"What is the weather like today?",
"Create a simple bar chart using Python",
"Convert 100 USD to EUR",
"What are the benefits of renewable energy?",
],
inputs=question_input,
label="๐ฌ Research & Analysis"
)
with gr.Column():
gr.Examples(
examples=[
"Analyze this image and describe what you see",
"Extract text from this image using OCR",
"Summarize the content of this document",
"Analyze the data in this CSV file",
"What insights can you find in this Excel file?",
],
inputs=question_input,
label="๐ File Analysis"
)
# Event handlers
def submit_question(question, history, files):
print(f"๐ฏ UI: Submit button clicked")
print(f"๐ UI: Question length: {len(question) if question else 0}")
print(f"๐ UI: Files count: {len(files) if files else 0}")
result_question, result_history = chatbot.process_question(question, history, files)
print(f"๐ UI: Returning results and clearing files")
return result_question, result_history, None # Clear files after processing
def clear_conversation():
print("๐งน UI: Clear conversation button clicked")
return chatbot.clear_history()
def clear_files():
print("๐๏ธ UI: Clear files button clicked")
return None
def export_conversation(history):
"""Export conversation history to a text file"""
print("๐พ UI: Export conversation button clicked")
if not history:
print("โ ๏ธ No conversation to export")
return None
try:
import tempfile
import datetime
# Create export content
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
export_content = f"# GAIA Lite Conversation Export\n"
export_content += f"Export Date: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
export_content += f"Total Messages: {len(history)}\n\n"
export_content += "=" * 50 + "\n\n"
for i, (user_msg, bot_msg) in enumerate(history, 1):
export_content += f"## Message {i}\n\n"
export_content += f"**User:** {user_msg}\n\n"
export_content += f"**Assistant:** {bot_msg}\n\n"
export_content += "-" * 30 + "\n\n"
# Save to temporary file
temp_file = tempfile.NamedTemporaryFile(
mode='w',
suffix=f'_gaia_lite_chat_export_{timestamp}.md',
delete=False,
encoding='utf-8'
)
temp_file.write(export_content)
temp_file.close()
print(f"๐ Conversation exported to: {temp_file.name}")
return temp_file.name
except Exception as e:
print(f"โ Error exporting conversation: {e}")
return None
# Connect the events
submit_btn.click(
fn=submit_question,
inputs=[question_input, chatbot_interface, file_upload],
outputs=[question_input, chatbot_interface, file_upload],
show_progress=True
)
question_input.submit(
fn=submit_question,
inputs=[question_input, chatbot_interface, file_upload],
outputs=[question_input, chatbot_interface, file_upload],
show_progress=True
)
clear_btn.click(
fn=clear_conversation,
outputs=[chatbot_interface],
show_progress=False
)
clear_files_btn.click(
fn=clear_files,
outputs=[file_upload],
show_progress=False
)
export_btn.click(
fn=export_conversation,
inputs=[chatbot_interface],
outputs=[download_file],
show_progress=True
)
return demo
if __name__ == "__main__":
print("\n" + "-"*50)
print("๐ Starting GAIA Lite...")
print("-"*50 + "\n")
# Create and launch the interface
demo = create_qna_interface()
demo.launch(
debug=True,
share=False,
server_name="0.0.0.0",
server_port=7860,
show_error=True
)