Spaces:
Sleeping
Sleeping
File size: 1,552 Bytes
d8a147c ad64507 12d1ae3 ecd405b 12d1ae3 ecd405b 12d1ae3 ecd405b 66aa8c3 5d53a0e 12d1ae3 66aa8c3 ad64507 11c8ace |
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 |
import tqdm
import gradio as gr
from huggingface_hub import InferenceClient
from gemini_api import get_travel_info, get_travel_recommendations
from pdf_convert import markdown_to_pdf_weasyprint
def generate_response(input_text):
user_query = input_text.strip()
if not user_query:
print("Please enter a valid travel query!")
return
travel_info = get_travel_info(user_query)
if travel_info and travel_info.get("to") and travel_info.get("from"):
travel_recommendations = get_travel_recommendations(
travel_info.get("intent"),
travel_info.get("from"),
travel_info.get("to"),
travel_info.get("departure_date"),
travel_info.get("duration"),
travel_info.get("budget"),
travel_info.get("num_people"),
)
pdf = markdown_to_pdf_weasyprint(travel_recommendations)
else:
print("Could not extract travel information. Please refine your query.")
return pdf
with gr.Blocks() as demo:
gr.Markdown("## PDF Processing with Gemini API")
message = "I want to plan a trip from Taiwan to Tokyo for 5 days with a budget of $10000 for 2 people, leaving on March 1st."
input_text = gr.Textbox(label="User Request", value=message, placeholder="Enter your travel detail here")
submit = gr.Button("Generate")
output_pdf = gr.File(label="Download Output PDF")
submit.click(fn=generate_response, inputs=input_text, outputs=output_pdf)
if __name__ == "__main__":
demo.launch(ssr_mode=False)
|