Rename app to app.py

app.py

@@ -0,0 +1,57 @@
+
+# Function to summarize text using BART
+def summarize_text(text):
+    inputs = tokenizer.encode("" + text, return_tensors="pt", max_length=1024, truncation=True)
+    summary_ids = model_bart.generate(inputs, max_length=150, min_length=30, length_penalty=2.0, num_beams=4, early_stopping=True)
+    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary
+
+def search_hotels(query_text, k=1):
+    try:
+        # Encode the query text
+        query_embedding = model.encode(query_text, convert_to_tensor=True)
+        query_embedding = query_embedding.cpu().numpy().reshape(1, -1)
+        query_embedding = normalize(query_embedding, norm='l2')
+
+        # Compute cosine similarity between query and stored embeddings
+        similarities = cosine_similarity(query_embedding, normalized_embeddings)
+
+        # Get indices of the top k similar hotels
+        top_indices = similarities[0].argsort()[-k:][::-1]
+
+        # Retrieve the top k similar hotels
+        top_hotels = df_copy_first_1000.iloc[top_indices]
+
+        # Prepare results
+        results = []
+        for _, row in top_hotels.iterrows():
+            # Create a summary for the description, review title, review text, and review count
+            summary_text = f"Description: {row['hotel_description']}\nReview Title: {row['review_title']}\nReview Text: {row['review_text']}\nReview Count: {row['review_count']}"
+            summary = summarize_text(summary_text)
+
+            result = (
+                f"Hotel Name: {row['hotel_name']}\n"
+                f"Locality: {row['locality']}\n"
+                f"Price Range: {row['price_range']}\n"
+                f"Rate: {row['rate']}\n"
+                f"\n {summary}\n"
+            )
+            results.append(result)
+
+        return "\n\n".join(results)
+
+    except Exception as e:
+        return f"An error occurred during the search: {e}"
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=search_hotels,
+    inputs=gr.Textbox(label="Enter your search query"),
+    outputs="text",
+    title="Hotel Search Engine",
+    description="Enter a query to search for hotels and get details about the top results."
+)
+
+# Launch Gradio Interface
+if __name__ == "__main__":
+    iface.launch()