Update app.py

app.py

@@ -1,169 +1,71 @@
-import gradio as gr
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import torch
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import gradio as gr
 
-# Step 1: Load the model and tokenizer from Hugging Face
-repo_id = "Rehman1603/fine_tune_gpt2_airline"  # Replace with your Hugging Face repo ID
-model = GPT2LMHeadModel.from_pretrained(repo_id)
-tokenizer = GPT2Tokenizer.from_pretrained(repo_id)
+# Load the fine-tuned model and tokenizer from Hugging Face
+model_name = "Rehman1603/Travel_fine_tuned_gpt2_model_final"  # Replace with your Hugging Face model path
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 
-# Step 2: Set the model to evaluation mode
-model.eval()
+# Move model to the appropriate device
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model.to(device)
 
-# Step 3: Define the dataset for filtering
-qa_data = [
-    ("What is AstroSynth?", "AstroSynth is an innovative program designed to synthesize oils from celestial materials."),
-    ("How does an airline manage its flight schedules?", "Airlines manage their flight schedules by using sophisticated software that takes into account factors such as demand, aircraft availability, airport slots, and regulatory requirements. The process involves planning, scheduling, and optimizing routes to maximize efficiency and profitability."),
-    ("What is the role of a load controller in an airline?", "A load controller ensures that the weight and balance of an aircraft are within safe limits. They coordinate with various departments to determine the optimal distribution of cargo, baggage, and passengers to ensure a safe and efficient flight."),
-    ("How do airlines handle overbooking?", "Airlines handle overbooking by offering incentives to passengers who voluntarily give up their seats. If no one volunteers, the airline may involuntarily deny boarding to some passengers and compensate them according to regulations."),
-    ("What is the purpose of a frequent flyer program?", "The purpose of a frequent flyer program is to reward loyal customers by offering points or miles for each flight taken. These points can be redeemed for free flights, upgrades, or other perks, encouraging repeat business."),
-    ("How do airlines calculate ticket prices?", "Airlines calculate ticket prices based on factors such as demand, competition, fuel costs, taxes, and the type of fare. Dynamic pricing models are often used to adjust prices in real-time based on market conditions."),
-    ("What is the difference between economy, business, and first-class cabins?", "Economy class offers standard seating and amenities, business class provides more comfort and services, and first class offers the highest level of luxury, including spacious seats, gourmet meals, and personalized service."),
-    ("How do airlines ensure safety during flight?", "Airlines ensure safety through rigorous maintenance schedules, pilot training, strict adherence to regulatory standards, and the use of advanced technology for navigation, communication, and monitoring systems."),
-    ("What is the role of air traffic control in an airline operation?", "Air traffic control ensures the safe and orderly flow of air traffic by providing pilots with instructions and information, managing flight paths, and coordinating with other aircraft and ground services."),
-    ("How do airlines handle baggage handling?", "Airlines handle baggage through a system of conveyor belts, scanners, and automated sorting systems. Baggage is tagged, tracked, and loaded onto the correct flight to ensure it reaches the destination with the passenger."),
-    ("What is the process for canceling a flight ticket?", "The process for canceling a flight ticket involves contacting the airline or travel agent, providing the necessary details, and following the airline's cancellation policy, which may include fees and refund options."),
-    ("How do airlines manage fuel efficiency?", "Airlines manage fuel efficiency through optimized flight paths, weight reduction, advanced engine technology, and operational practices such as reducing taxi times and using ground power instead of auxiliary power units."),
-    ("What is the role of a flight attendant?", "A flight attendant's role is to ensure the safety and comfort of passengers during the flight. They provide assistance, serve meals and beverages, and respond to emergencies as needed."),
-    ("How do airlines handle delays and cancellations?", "Airlines handle delays and cancellations by providing passengers with updated information, rebooking options, and compensation according to regulations. They also work to minimize disruptions through contingency planning."),
-    ("What is the purpose of a black box in an aircraft?", "The black box, or flight data recorder, is used to record flight data and cockpit conversations. It helps investigators determine the cause of an accident or incident by providing detailed information about the flight."),
-    ("How do airlines ensure customer satisfaction?", "Airlines ensure customer satisfaction by providing high-quality service, addressing complaints promptly, offering loyalty programs, and continuously improving their products and services based on customer feedback."),
-    ("What is the role of a ground crew in an airline?", "The ground crew handles various tasks such as aircraft cleaning, fueling, baggage loading, and passenger assistance. They ensure that the aircraft is ready for departure and provide support during boarding and deplaning."),
-    ("How do airlines manage partnerships and alliances?", "Airlines manage partnerships and alliances by collaborating with other carriers to expand their network, share resources, and offer seamless travel experiences to passengers through codesharing and reciprocal benefits."),
-    ("What is the process for checking in for a flight?", "The process for checking in for a flight involves providing identification and booking details at the airport counter, kiosk, or online. Passengers receive their boarding pass and baggage tags, and their luggage is checked in."),
-    ("How do airlines handle in-flight entertainment?", "Airlines handle in-flight entertainment by offering a variety of options such as movies, TV shows, music, games, and Wi-Fi. The content is provided through seatback screens, personal devices, or streaming services."),
-    ("What is the role of a pilot in an airline?", "The pilot's role is to safely operate the aircraft, navigate to the destination, and ensure the safety of passengers and crew. They are responsible for pre-flight checks, takeoff, landing, and managing in-flight operations."),
-    ("How do airlines manage cargo operations?", "Airlines manage cargo operations by using dedicated cargo aircraft or passenger aircraft with cargo holds. They handle freight booking, loading, tracking, and delivery to ensure timely and secure transportation of goods."),
-    ("What is the purpose of a boarding pass?", "The boarding pass serves as proof of a passenger's reservation and authorization to board the flight. It contains details such as the passenger's name, flight number, seat assignment, and boarding gate."),
-    ("How do airlines handle passenger complaints?", "Airlines handle passenger complaints by providing channels for feedback, such as customer service hotlines, email, or in-person assistance. They investigate the issue, offer solutions, and work to improve their services based on complaints."),
-    ("What is the role of a flight dispatcher?", "A flight dispatcher assists in planning and monitoring flights. They provide pilots with flight plans, weather updates, and other critical information to ensure a safe and efficient operation."),
-    ("How do airlines manage airport slots?", "Airlines manage airport slots by applying for and securing permission to operate at specific times and gates. Slot allocation is based on factors such as demand, airport capacity, and regulatory requirements."),
-    ("What is the process for changing a flight ticket?", "The process for changing a flight ticket involves contacting the airline or travel agent, providing the necessary details, and following the airline's change policy, which may include fees and fare differences."),
-    ("How do airlines handle in-flight meals?", "Airlines handle in-flight meals by partnering with caterers to prepare and deliver meals based on the flight duration, route, and passenger preferences. Meals are served by flight attendants during the flight."),
-    ("What is the role of a baggage handler?", "A baggage handler's role is to load and unload baggage from the aircraft, ensuring it is correctly tagged and sorted. They also handle lost or delayed baggage claims and coordinate with other ground staff."),
-    ("How do airlines manage loyalty program partnerships?", "Airlines manage loyalty program partnerships by collaborating with hotels, car rental companies, and other businesses to offer points-earning and redemption opportunities across multiple industries, enhancing the value of the program."),
-    ("What is the purpose of a flight plan?", "The purpose of a flight plan is to provide detailed information about the flight, including the route, altitude, speed, and fuel requirements. It is used by pilots and air traffic control to ensure safe and efficient operations."),
-    ("How do airlines handle passenger data privacy?", "Airlines handle passenger data privacy by adhering to regulations such as GDPR and CCPA. They implement data protection measures, secure storage systems, and provide transparency about how data is collected and used."),
-    ("What is the role of a gate agent?", "A gate agent's role is to assist passengers with boarding and deplaning, verify boarding passes, and manage the boarding process. They also handle any last-minute changes or issues related to the flight."),
-    ("How do airlines manage aircraft maintenance?", "Airlines manage aircraft maintenance through scheduled checks, inspections, and repairs. They follow strict regulatory standards and use advanced technology to ensure the aircraft remains airworthy and safe."),
-    ("What is the purpose of a crew briefing?", "The purpose of a crew briefing is to provide pilots, flight attendants, and other crew members with essential information about the flight, including safety procedures, weather conditions, and any special instructions."),
-    ("How do airlines handle passenger health and safety during the pandemic?", "Airlines handle passenger health and safety during the pandemic by implementing measures such as mask mandates, enhanced cleaning protocols, air filtration systems, social distancing, and health screenings."),
-    ("What is the role of a flight engineer?", "A flight engineer's role is to monitor and manage the aircraft's systems, including engines, fuel, and navigation. They assist the pilot in ensuring the aircraft operates safely and efficiently, especially on older or larger planes."),
-    ("How do airlines manage seasonal demand fluctuations?", "Airlines manage seasonal demand fluctuations by adjusting flight schedules, adding or reducing capacity, and offering promotional fares. They also use data analytics to predict demand and optimize operations accordingly."),
-    ("What is the purpose of a baggage claim ticket?", "The purpose of a baggage claim ticket is to provide proof of ownership and tracking for checked luggage. It helps passengers identify their bags and assists airlines in locating lost or delayed baggage."),
-    ("How do airlines handle passenger rebooking after a cancellation?", "Airlines handle passenger rebooking after a cancellation by offering alternative flight options, waiving change fees, and providing compensation according to regulations. They prioritize customer convenience and minimize disruptions."),
-    ("What is the role of a flight operations manager?", "A flight operations manager oversees the planning and execution of flight operations. They ensure compliance with regulations, manage crew schedules, and coordinate with other departments to maintain efficiency and safety."),
-    ("How do airlines manage aircraft turnaround time?", "Airlines manage aircraft turnaround time by optimizing ground operations, such as baggage loading, cleaning, and refueling. Efficient coordination between ground staff and crew helps minimize delays between flights."),
-    ("What is the purpose of a crew rest facility?", "The purpose of a crew rest facility is to provide pilots and flight attendants with a place to rest and recover during long flights. These facilities ensure that crew members remain alert and fit to operate the aircraft safely."),
-    ("How do airlines handle passenger seating assignments?", "Airlines handle passenger seating assignments by offering options for pre-selection, assigning seats based on fare class, or allowing passengers to choose seats at check-in. They also manage upgrades and special requests, such as extra legroom or companion seating."),
-    ("What is the role of a flight simulator?", "A flight simulator is used for pilot training and testing. It provides a realistic environment to practice flying, handle emergencies, and familiarize themselves with aircraft systems without the risks of actual flight."),
-    ("How do airlines manage cargo security?", "Airlines manage cargo security by implementing strict screening processes, using advanced scanning technology, and adhering to international regulations. They also work with security agencies to prevent unauthorized or hazardous materials from being transported."),
-    ("What is the purpose of a crew scheduling system?", "The purpose of a crew scheduling system is to optimize the assignment of pilots and flight attendants to flights. It ensures compliance with labor regulations, maximizes efficiency, and minimizes costs while maintaining safety standards."),
-    ("How do airlines handle passenger check-in kiosks?", "Airlines handle passenger check-in kiosks by providing self-service options that allow passengers to print boarding passes, select seats, and check luggage. These kiosks reduce wait times and streamline the check-in process."),
-    ("What is the role of a flight data analyst?", "A flight data analyst collects and analyzes data related to flight operations, such as performance, delays, and fuel consumption. They provide insights to improve efficiency, safety, and profitability for the airline."),
-    ("How do airlines manage passenger boarding?", "Airlines manage passenger boarding by using strategies such as zone boarding, pre-boarding for special needs passengers, and mobile boarding passes. Efficient boarding processes help reduce delays and improve the overall travel experience."),
-    ("What is the purpose of a crew briefing room?", "The purpose of a crew briefing room is to provide a space for pilots and flight attendants to receive pre-flight instructions, review safety procedures, and discuss any special requirements for the upcoming flight."),
-    ("How do airlines handle passenger feedback?", "Airlines handle passenger feedback by collecting data through surveys, social media, and customer service channels. They analyze feedback to identify areas for improvement and implement changes to enhance the passenger experience."),
-    ("What is the role of a flight operations coordinator?", "A flight operations coordinator assists in planning and monitoring flight schedules. They ensure that all operational requirements are met, coordinate with various departments, and provide support to the flight crew."),
-    ("How do airlines manage aircraft cleaning?", "Airlines manage aircraft cleaning by using specialized teams to perform deep cleans between flights. They follow strict hygiene protocols, use eco-friendly cleaning products, and ensure compliance with health and safety regulations."),
-    ("What is the purpose of a crew rest period?", "The purpose of a crew rest period is to ensure that pilots and flight attendants are well-rested and fit to operate the aircraft safely. Rest periods are regulated by aviation authorities to prevent fatigue and maintain safety standards."),
-    ("How do airlines handle passenger upgrades?", "Airlines handle passenger upgrades by offering them through loyalty programs, last-minute promotions, or as a reward for frequent flyers. Upgrades may include moving to business or first class for enhanced comfort and services."),
-    ("What is the role of a flight operations director?", "A flight operations director oversees all aspects of flight operations, including scheduling, safety, and compliance. They ensure that the airline meets its operational goals while maintaining high safety and service standards."),
-    ("How do airlines manage aircraft refueling?", "Airlines manage aircraft refueling by coordinating with ground service providers and fuel suppliers. They ensure that the correct amount of fuel is loaded based on flight distance, weather conditions, and regulatory requirements."),
-    ("What is the purpose of a crew rest area?", "The purpose of a crew rest area is to provide a comfortable and secure space for pilots and flight attendants to rest during long flights. These areas are designed to ensure that crew members remain alert and fit to operate the aircraft safely."),
-    ("How do airlines handle passenger special requests?", "Airlines handle passenger special requests by offering options such as dietary accommodations, wheelchair assistance, or seating preferences. They work to fulfill these requests to enhance the passenger experience and ensure comfort."),
-    ("What is the role of a flight operations analyst?", "A flight operations analyst reviews and analyzes data related to flight operations, such as performance metrics, delays, and fuel consumption. They provide insights to improve efficiency, safety, and profitability for the airline."),
-    ("How do airlines manage passenger baggage tracking?", "Airlines manage passenger baggage tracking by using barcode or RFID tags, automated sorting systems, and real-time monitoring. These technologies help ensure that baggage is loaded onto the correct flight and reaches the destination with the passenger."),
-    ("What is the purpose of a crew scheduling manager?", "The purpose of a crew scheduling manager is to oversee the assignment of pilots and flight attendants to flights. They ensure compliance with labor regulations, maximize efficiency, and minimize costs while maintaining safety standards."),
-    ("How do airlines handle passenger check-in queues?", "Airlines handle passenger check-in queues by offering self-service kiosks, mobile check-in options, and dedicated staff to assist passengers. Efficient queue management helps reduce wait times and improve the overall travel experience."),
-    ("What is the role of a flight operations specialist?", "A flight operations specialist assists in planning and monitoring flight schedules. They ensure that all operational requirements are met, coordinate with various departments, and provide support to the flight crew."),
-    ("How do airlines manage aircraft de-icing?", "Airlines manage aircraft de-icing by using specialized equipment and chemicals to remove ice and snow from the aircraft's surface. This process ensures safe takeoff and prevents ice buildup that could affect flight performance."),
-    ("What is the purpose of a crew briefing checklist?", "The purpose of a crew briefing checklist is to ensure that all essential information is communicated to the flight crew before departure. It includes safety procedures, weather updates, and any special instructions for the flight."),
-    ("How do airlines handle passenger health screenings?", "Airlines handle passenger health screenings by implementing measures such as temperature checks, health declaration forms, and contact tracing. These measures help ensure the safety of passengers and crew during the pandemic."),
-    ("What is the role of a flight operations assistant?", "A flight operations assistant supports the flight operations team by handling administrative tasks, coordinating with various departments, and providing logistical support. They ensure that all operational requirements are met efficiently."),
-    ("How do airlines manage aircraft parking?", "Airlines manage aircraft parking by coordinating with airport authorities to secure appropriate gates and parking spots. They ensure that aircraft are parked safely and efficiently, minimizing delays and disruptions."),
-    ("What is the purpose of a crew rest policy?", "The purpose of a crew rest policy is to ensure that pilots and flight attendants are well-rested and fit to operate the aircraft safely. These policies are regulated by aviation authorities to prevent fatigue and maintain safety standards."),
-    ("How do airlines handle passenger seating changes?", "Airlines handle passenger seating changes by offering options for seat selection, upgrades, or reassignments. They work to accommodate special requests and ensure that passengers are seated in a manner that maximizes comfort and safety."),
-    ("What is the role of a flight operations coordinator?", "A flight operations coordinator assists in planning and monitoring flight schedules. They ensure that all operational requirements are met, coordinate with various departments, and provide support to the flight crew."),
-    ("How do airlines manage aircraft maintenance scheduling?", "Airlines manage aircraft maintenance scheduling by using advanced software to plan and track maintenance activities. They ensure that all checks and repairs are performed on time, maintaining the aircraft's airworthiness and safety.")
-]
-
-# Step 4: Define a function to generate answers
-def ask_question(question):
-    # Check if the question is in the dataset
-    for q, a in qa_data:
-        if question.lower() == q.lower():
-            return a
-    
-    # If the question is not in the dataset, generate a response using the model
+# Function to generate responses
+def generate_response(question, max_length=1024, temperature=0.5, top_k=50, top_p=0.95):
     input_text = f"Question: {question} Answer:"
-    input_ids = tokenizer.encode(input_text, return_tensors="pt")
+    input_ids = tokenizer.encode(input_text, return_tensors="pt").to(device)
     
-    with torch.no_grad():
-        output = model.generate(
-            input_ids,
-            max_length=1024,
-            num_return_sequences=1,
-            no_repeat_ngram_size=2,
-            do_sample=True,
-            top_k=50,
-            top_p=0.95,
-            temperature=0.7
-        )
-    
-    answer = tokenizer.decode(output[0], skip_special_tokens=True)
-    answer = answer.split("Answer:")[-1].strip()
-    
-    # If the model fails to generate a meaningful response, return a default message
-    if not answer or answer.lower() == "answer:":
-        return "I'm sorry, but I don't have information about that. I can only provide information related to airline systems."
+    # Generate response
+    output = model.generate(
+        input_ids,
+        max_length=max_length,
+        temperature=temperature,
+        #num_return_sequences=1,
+        #no_repeat_ngram_size=2,
+        top_k=top_k,
+        top_p=top_p,
+        do_sample=True,
+        pad_token_id=tokenizer.eos_token_id
+    )
     
-    return answer
+    # Decode and return the response
+    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    return response.split("Answer:")[-1].strip()
 
-# Step 5: Create the Gradio interface
-def chat_interface(input_text, chat_history=[]):
-    # Handle greetings
-    greetings = ["hi", "hello", "aslam o alikum"]
-    if input_text.strip().lower() in greetings:
-        bot_response = "Hello! How can I assist you with airline-related questions?"
-    else:
-        # Handle other questions
-        bot_response = ask_question(input_text)
-    
-    # Update chat history
-    chat_history.append((input_text, bot_response))
-    
-    # Return updated chat history
-    return chat_history, chat_history
+# Gradio Interface
+def chat_interface(question, history):
+    response = generate_response(question)
+    history = history or []
+    history.append((question, response))
+    return history, history
 
-# Step 6: Launch the Gradio app
-with gr.Blocks() as iface:
-    gr.Markdown("# Airline Chatbot\nAsk me anything about airlines!")
-    
-    # Chatbot component
-    chat_history = gr.Chatbot(label="Chatbot")
-    
-    # Input textbox and send button
+# Test questions
+test_questions = [
+    "Hi",
+    "What is the duration of the Economy Umrah Package?",
+    "Good morning",
+    "What is the distance of hotels from Haram in Package 4?",
+    "What is the price of a 14-night Umrah package with air tickets in Package 1",
+    "What is the price of a 20-night Umrah package without air tickets?"
+]
+
+# Create Gradio interface
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    state = gr.State()
+    with gr.Row():
+        txt = gr.Textbox(show_label=False, placeholder="Enter your question here")
     with gr.Row():
-        user_input = gr.Textbox(label="You", placeholder="Type your message here...", lines=2)
-        send_button = gr.Button("Send")
+        submit_btn = gr.Button("Submit")
+        clear_btn = gr.Button("Clear")
     
-    # Examples
-    gr.Examples(
-        examples=[
-            "How do airlines manage fuel efficiency?",
-            "How does an airline manage its flight schedules?",
-            "How do airlines handle passenger seating changes?",
-            "What is the purpose of a crew rest area?"
-        ],
-        inputs=user_input
-    )
+    # Example questions
+    examples = gr.Examples(examples=test_questions, inputs=txt)
     
-    # Connect the send button to the chat interface
-    send_button.click(
-        fn=chat_interface,
-        inputs=[user_input, chat_history],
-        outputs=[chat_history, chat_history]
-    )
+    # Event handling
+    submit_btn.click(chat_interface, [txt, state], [chatbot, state])
+    clear_btn.click(lambda: None, None, chatbot, queue=False)
 
-iface.launch(debug=True)
+# Launch the interface
+demo.launch(debug=True)