Create app.py

app.py

@@ -0,0 +1,50 @@
+import gradio as gr
+from llm_utils import process_natural_language_query
+from data import flights_db, users_db, bookings_db
+import uuid
+import datetime
+
+def search_flights(query):
+    criteria = process_natural_language_query(query)
+    results = [
+        f for f in flights_db
+        if f['source'].lower() == criteria['source'].lower()
+        and f['destination'].lower() == criteria['destination'].lower()
+        and f['date'] == criteria['date']
+    ]
+    return results
+
+def book_flight(user_id, flight_id):
+    user = next((u for u in users_db if u['id'] == user_id), None)
+    flight = next((f for f in flights_db if f['id'] == flight_id), None)
+    if not user or not flight:
+        return "❌ User or flight not found."
+    booking_id = str(uuid.uuid4())
+    bookings_db.append({
+        'id': booking_id,
+        'user_id': user_id,
+        'flight_id': flight_id,
+        'timestamp': str(datetime.datetime.now())
+    })
+    return f"✅ Booking confirmed! ID: {booking_id}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("## ✈️ Secure Flight Booking System with LLM")
+
+    with gr.Row():
+        query = gr.Textbox(label="Enter your flight query")
+        search_btn = gr.Button("Search Flights")
+
+    results = gr.JSON(label="Available Flights")
+
+    with gr.Row():
+        user_id = gr.Textbox(label="User ID")
+        flight_id = gr.Textbox(label="Flight ID")
+        book_btn = gr.Button("Book Flight")
+
+    booking_output = gr.Textbox(label="Booking Status")
+
+    search_btn.click(fn=search_flights, inputs=query, outputs=results)
+    book_btn.click(fn=book_flight, inputs=[user_id, flight_id], outputs=booking_output)
+
+demo.launch()