Spaces:
Sleeping
Sleeping
File size: 5,715 Bytes
54ee5d5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
import torch
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM # Fixed import name
import gradio as gr
import warnings
warnings.filterwarnings('ignore')
class SpaceBot:
def __init__(self):
# Initialize model and tokenizer
print("Loading model and tokenizer...")
self.model_name = "google/flan-t5-base" # Using a more stable model
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
# Space facts database
self.space_facts = {
"apollo": [
"The Apollo program ran from 1961 to 1972.",
"Apollo 11 was the first mission to land humans on the Moon.",
"Neil Armstrong and Buzz Aldrin walked on the Moon on July 20, 1969.",
"The Apollo program included 17 space missions in total."
],
"iss": [
"The International Space Station orbits Earth at about 400 kilometers above the surface.",
"The ISS has been continuously occupied since November 2000.",
"The ISS travels at approximately 7.66 kilometers per second.",
"The space station is about the size of a football field."
],
"mars": [
"Mars is often called the Red Planet due to its reddish appearance.",
"NASA's Perseverance rover landed on Mars in 2021.",
"Mars has two small moons: Phobos and Deimos.",
"The first successful Mars landing was by NASA's Viking 1 in 1976."
],
"telescope": [
"The Hubble Space Telescope was launched in 1990.",
"The James Webb Space Telescope is the largest space telescope ever built.",
"Space telescopes can see more clearly than ground-based telescopes.",
"The Hubble has made over 1.5 million observations."
],
"general": [
"The first human in space was Yuri Gagarin in 1961.",
"SpaceX achieved the first successful landing of a reusable rocket in 2015.",
"There are currently over 5,000 satellites orbiting Earth.",
"Light takes about 8 minutes to travel from the Sun to Earth."
]
}
def get_relevant_facts(self, query):
"""Get relevant facts based on keywords in the query"""
query = query.lower()
relevant_facts = []
# Check each category for relevant facts
for category, facts in self.space_facts.items():
if category in query or any(keyword in query for keyword in category.split()):
relevant_facts.extend(facts)
# If no specific facts found, return general facts
if not relevant_facts:
relevant_facts = self.space_facts["general"]
return " ".join(relevant_facts[:2]) # Return two relevant facts
def generate_response(self, query):
try:
# Get relevant facts as context
context = self.get_relevant_facts(query)
# Prepare input text
input_text = f"Using this context: {context}\nAnswer this question: {query}"
# Tokenize input
inputs = self.tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
# Generate response
outputs = self.model.generate(
inputs.input_ids,
max_length=150,
min_length=30,
num_beams=4,
temperature=0.7,
no_repeat_ngram_size=2
)
# Decode response
response = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
return response
except Exception as e:
print(f"Error generating response: {str(e)}")
return "I apologize, but I couldn't generate a proper response. Please try asking your question in a different way."
def chat_with_bot(message, history):
"""Handle chat interactions"""
try:
response = bot.generate_response(message)
return [(response, None)]
except Exception as e:
print(f"Error in chat interface: {str(e)}")
return [("I encountered an error. Please try asking your question again.", None)]
print("Initializing SpaceBot...")
# Initialize the bot
bot = SpaceBot()
print("SpaceBot initialization complete!")
# Create Gradio interface
with gr.Blocks(theme=gr.themes.Soft()) as demo:
gr.Markdown("""
# 🚀 Space Explorer Chat
Ask me anything about space exploration and astronomy!
""")
chatbot = gr.ChatInterface(
chat_with_bot,
chatbot=gr.Chatbot(height=400),
textbox=gr.Textbox(
placeholder="Ask about space missions, planets, or astronomy...",
container=False
),
title="Space Explorer",
description="Your guide to space exploration and astronomy",
theme="soft",
examples=[
"What was the Apollo program?",
"Tell me about the International Space Station",
"What do you know about Mars exploration?",
"Tell me about space telescopes",
"What are some interesting space facts?"
],
)
gr.Markdown("""
### Topics I can help with:
- Space missions and history
- Planets and astronomy
- Space stations and satellites
- Space telescopes
- General space facts
""")
# Launch the interface
demo.launch(share=True) |