Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from sentence_transformers import SentenceTransformer, util | |
| import openai | |
| import os | |
| #trying so when user puts la or los angles specific pic comes: | |
| # URL or path to your image file | |
| #PICTURE_URL = "Stars/sf.png" | |
| #def respond(user_input): | |
| # if "los angeles" in user_input.lower() or "la" in user_input.lower(): | |
| #return f"Here's a picture of Los Angeles!", PICTURE_URL | |
| # else: | |
| # return "How can I help you with astronomy?", None | |
| # Define the Gradio interface | |
| #iface = gr.Interface( | |
| # fn=respond, | |
| # inputs="text", | |
| # outputs=["text", "image"] | |
| #) | |
| # Launch the interface | |
| #iface.launch() | |
| os.environ["TOKENIZERS_PARALLELISM"] = "false" | |
| # Initialize paths and model identifiers for easy configuration and maintenance | |
| filename = "output_topic_details.txt" # Path to the file storing chess-specific details | |
| retrieval_model_name = 'output/sentence-transformer-finetuned/' | |
| # Define paths to images | |
| path_to_sf_image = "Stars/sf.png" | |
| path_to_sacramento_image = "Stars/sacramento.png" | |
| path_to_la_image = "Stars/la.png" | |
| openai.api_key = os.environ["OPENAI_API_KEY"] | |
| system_message = "You are an astronomy chatbot named Starfinder specialized in providing information on stargazing, astronomical events, and outer space." | |
| # Initial system message to set the behavior of the assistant | |
| messages = [{"role": "system", "content": system_message}] | |
| # Attempt to load the necessary models and provide feedback on success or failure | |
| try: | |
| retrieval_model = SentenceTransformer(retrieval_model_name) | |
| print("Models loaded successfully.") | |
| except Exception as e: | |
| print(f"Failed to load models: {e}") | |
| def load_and_preprocess_text(filename): | |
| """ | |
| Load and preprocess text from a file, removing empty lines and stripping whitespace. | |
| """ | |
| try: | |
| with open(filename, 'r', encoding='utf-8') as file: | |
| segments = [line.strip() for line in file if line.strip()] | |
| print("Text loaded and preprocessed successfully.") | |
| return segments | |
| except Exception as e: | |
| print(f"Failed to load or preprocess text: {e}") | |
| return [] | |
| segments = load_and_preprocess_text(filename) | |
| def find_relevant_segment(user_query, segments): | |
| """ | |
| Find the most relevant text segment for a user's query using cosine similarity among sentence embeddings. | |
| This version finds the best match based on the content of the query. | |
| """ | |
| try: | |
| # Lowercase the query for better matching | |
| lower_query = user_query.lower() | |
| # Encode the query and the segments | |
| query_embedding = retrieval_model.encode(lower_query) | |
| segment_embeddings = retrieval_model.encode(segments) | |
| # Compute cosine similarities between the query and the segments | |
| similarities = util.pytorch_cos_sim(query_embedding, segment_embeddings)[0] | |
| # Find the index of the most similar segment | |
| best_idx = similarities.argmax() | |
| # Return the most relevant segment | |
| return segments[best_idx] | |
| except Exception as e: | |
| print(f"Error in finding relevant segment: {e}") | |
| return "" | |
| def generate_response(user_query, relevant_segment): | |
| """ | |
| Generate a response emphasizing the bot's capability in providing astronomical information. | |
| """ | |
| try: | |
| user_message = f"Here's the information on outer space: {relevant_segment}" | |
| # Append user's message to messages list | |
| messages.append({"role": "user", "content": user_message}) | |
| response = openai.ChatCompletion.create( | |
| model="gpt-3.5-turbo", | |
| messages=messages, | |
| max_tokens=150, | |
| temperature=0.2, | |
| top_p=1, | |
| frequency_penalty=0, | |
| presence_penalty=0 | |
| ) | |
| # Extract the response text | |
| output_text = response['choices'][0]['message']['content'].strip() | |
| # Append assistant's message to messages list for context | |
| messages.append({"role": "assistant", "content": output_text}) | |
| return output_text | |
| except Exception as e: | |
| print(f"Error in generating response: {e}") | |
| return f"Error in generating response: {e}" | |
| # below perfect code | |
| def query_model(question): | |
| """ | |
| Process a question, find relevant information, and generate a response. | |
| """ | |
| if question == "": | |
| return "Welcome to Starfinder! Ask me anything about outer space, stargazing, and upcoming astronomical events.", None | |
| if "san francisco" in question.lower(): | |
| return "There are many locations near San Francisco where you can stargaze: Lick Observatory (Mount Hamilton), Chabot Space & Science Center (Oakland) , Twin Peaks (SF), Sibley Volcanic National Reserve (Oakland), Mount Tamalpais (Marin), San Francisco State University Observatory (SF), Mount Diablo (East Bay)!", "https://huggingface.co/spaces/Starfinders/Stars/resolve/main/SF%20CANVA" | |
| if "sacramento" in question.lower(): | |
| return "There are many locations near Sacramento where you can stargaze: Kalithea Park, Northstar Park, Curtis Park, Lake Theodore, Casa Bella Verde, McKinley Park, Tiscornia Park, Old Sacramento Waterfront.", "https://huggingface.co/spaces/Starfinders/Stars/resolve/main/SACROMENTO%20CANVA" | |
| if "los angeles" in question.lower() or "la" in question.lower(): | |
| return "There are many locations near Los Angeles where you can stargaze: Leo Carrillo State Beach (Malibu), Malibu Creek State Park (Malibu), Griffith Observatory (Griffith Park), Mount Wilson Observatory (Angeles Crest)", "https://huggingface.co/spaces/Starfinders/Stars/resolve/main/LOS%20ANGELES%20CANVA.png" | |
| # above perfect code | |
| if "Orion" in question.lower(): | |
| return "Best viewed from January to March in the Northern Hemisphere; known for the iconic Orion's Belt, consisting of the three stars Alnitak, Alnilam, and Mintaka, and is flanked by the bright stars Betelgeuse and Rigel. Orion is often associated with the Greek myth of the great hunter who was placed among the stars by Zeus.", "https://huggingface.co/spaces/Starfinders/Stars/resolve/main/SF%20CANVAhttps://huggingface.co/spaces/Starfinders/Stars/resolve/main/orion%20.png" | |
| if "Ursa Major"in question.lower(): | |
| return "Visible year-round in the Northern Hemisphere; home to the Big Dipper, an asterism used for navigation and to find Polaris. In mythology, Ursa Major represents Callisto, a nymph who was transformed into a bear by Zeus and placed in the sky.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/ursa%20major.png" | |
| if "Ursa Minor" in question.lower(): | |
| return "Also visible year-round in the Northern Hemisphere; contains Polaris, the current North Star, which has been a vital navigational tool throughout history. It represents Arcas, the son of Callisto, who was also turned into a bear and placed near his mother in the sky.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/ursa%20minor.png" | |
| if "Cassiopeia" in question.lower(): | |
| return " Best seen in the autumn months; notable for its distinctive W shape, which represents the vain queen Cassiopeia from Greek mythology. She was punished by being placed in the sky, where she circles the North Star, sometimes appearing upside-down.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/Cassiopeia%20.png" | |
| if "Cyngnus" in question.lower(): | |
| return "Most visible in the summer; contains the Northern Cross asterism and the bright star Deneb, which forms part of the Summer Triangle. Cygnus, the Swan, is often associated with the story of Zeus transforming into a swan or the tale of Phaethon, who fell into the Eridanus river.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/Cygnus.png" | |
| if "Scropius" in question.lower(): | |
| return "Best seen in July and August; easily recognized by the bright red star Antares at its heart and its curved tail of stars. Scorpius is linked to the myth of the scorpion sent by Gaia to kill Orion, which explains why they are placed on opposite sides of the sky.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/SCORPIUS.png" | |
| if "Leo" in question.lower(): | |
| return "Best viewed in April; known for the sickle-shaped pattern representing the lion's mane and the bright star Regulus, which is known as the Heart of the Lion. Leo is associated with the Nemean Lion, a beast slain by Hercules as one of his twelve labors.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/leo.png" | |
| if "Taurus" in question.lower(): | |
| return "Best visible in November and December; characterized by the bright red star Aldebaran and the nearby Pleiades star cluster, also known as the Seven Sisters. In mythology, Taurus represents the bull form of Zeus, who abducted Europa.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/Taurus%20.png" | |
| if "Gemini" in question.lower(): | |
| return "Best seen in February and March; features the twin stars Castor and Pollux, named after the mythological twins who were granted shared immortality by Zeus. The constellation is often associated with companionship and duality.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/Gemini.png" | |
| if "Libra" in question.lower(): | |
| return "Best viewed in June; symbolizes the scales of justice and balance, often associated with the Greek goddess Astraea. Libra is one of the few constellations representing an inanimate object, highlighting its unique role as a symbol of equilibrium and fairness.","https://huggingface.co/spaces/Starfinders/Stars/resolve/main/Libra.png" | |
| #below perfect code | |
| relevant_segment = find_relevant_segment(question, segments) | |
| if not relevant_segment: | |
| return "Could not find specific information. Please refine your question.", None | |
| response = generate_response(question, relevant_segment) | |
| return response, None | |
| # Define the welcome message and specific topics the chatbot can provide information about | |
| welcome_message = """ | |
| # ✧ Welcome to Starfinder! | |
| ## Your AI-driven assistant for all astronomy-related queries. Created by Aarna, Aditi, and Anastasia of the 2024 Kode With Klossy SF Camp. | |
| ## Please ask StarFinder questions about SF, LA, Sacromento | |
| """ | |
| topics = """ | |
| ### Feel Free to ask me anything from the topics below! | |
| - The Night sky | |
| - Outer space insights | |
| - Light pollution | |
| - Stargazing spots | |
| - Celestial events | |
| - Astronomy tips | |
| """ | |
| def display_title(): | |
| return "https://huggingface.co/spaces/Starfinders/Stars/resolve/main/logo.png" | |
| # Setup the Gradio Blocks interface with custom layout components | |
| with gr.Blocks(theme='earneleh/paris') as demo: | |
| gr.Image(display_title(), show_label = False, show_share_button = False, show_download_button = False, width = 150, height = 150) | |
| gr.Markdown(welcome_message) # Display the formatted welcome message | |
| with gr.Row(): | |
| with gr.Column(): | |
| gr.Markdown(topics) # Show the topics on the left side | |
| with gr.Row(): | |
| with gr.Column(): | |
| question = gr.Textbox(label="Your question", placeholder="What do you want to ask about?") | |
| answer = gr.Textbox(label="StarFinder Response", placeholder="StarFinder will respond here...", interactive=False, lines=10) | |
| image_output = gr.Image(label="Image Output") # Add an Image component | |
| submit_button = gr.Button("Submit") | |
| submit_button.click(fn=query_model, inputs=question, outputs=[answer, image_output]) # Update outputs to include the image component | |
| # Launch the Gradio app to allow user interaction | |
| demo.launch(share=True) | |