gamingflexer commited on
Commit
4b0bd4a
·
1 Parent(s): b33c42d

Add API integration for product search and details

Browse files
Files changed (1) hide show
  1. src/app2.py +35 -0
src/app2.py CHANGED
@@ -1,10 +1,32 @@
1
  import gradio as gr
2
  import pandas as pd
3
  import os
 
4
 
5
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
6
  SEVER_IP = os.environ.get("SEVER_IP")
7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  def sample_fun(voice_input, prodcut_id):
9
  print(prodcut_id)
10
  return
@@ -16,8 +38,21 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=gr.themes.colors.red, seconda
16
  prodcut_id = gr.Textbox(label="Enter Product ID")
17
  with gr.Row():
18
  submit_button_tab_1 = gr.Button("Start")
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
 
21
  submit_button_tab_1.click(fn=sample_fun,inputs=[voice_input,prodcut_id])
 
22
 
23
  demo.launch()
 
1
  import gradio as gr
2
  import pandas as pd
3
  import os
4
+ import requests
5
 
6
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
7
  SEVER_IP = os.environ.get("SEVER_IP")
8
 
9
+ def get_total_number_of_products():
10
+ response = requests.get(f'{SEVER_IP}/api/total_number_of_products/')
11
+ if response.status_code == 200:
12
+ return response.json()['total_number_of_products']
13
+ else:
14
+ return "Error fetching total number of products"
15
+
16
+ def search_products(product_name):
17
+ response = requests.get(f'{SEVER_IP}/api/search_products/', params={'name': product_name})
18
+ if response.status_code == 200:
19
+ return pd.DataFrame(response.json())
20
+ else:
21
+ return pd.DataFrame([]) # Return an empty DataFrame in case of an error
22
+
23
+ def get_product_details_by_id(product_id):
24
+ response = requests.get(f'{SEVER_IP}/api/product_details/{product_id}/')
25
+ if response.status_code == 200:
26
+ return response.json() # Returns the product details as a dictionary
27
+ else:
28
+ return {"error": f"Product with ID {product_id} not found or error occurred."}
29
+
30
  def sample_fun(voice_input, prodcut_id):
31
  print(prodcut_id)
32
  return
 
38
  prodcut_id = gr.Textbox(label="Enter Product ID")
39
  with gr.Row():
40
  submit_button_tab_1 = gr.Button("Start")
41
+
42
+ with gr.Tab("Search Catalog"):
43
+ with gr.Row():
44
+ total_no_of_products = gr.Textbox(value=str(get_total_number_of_products()),label="Total Products")
45
+ with gr.Row():
46
+ embbed_text_search = gr.Textbox(label="Enter Product Name")
47
+ submit_button_tab_4 = gr.Button("Start")
48
+ dataframe_output_tab_4 = gr.Dataframe(headers=['id', 'barcode', 'brand', 'sub_brand', 'manufactured_by', 'product_name',
49
+ 'weight', 'variant', 'net_content', 'price', 'parent_category',
50
+ 'child_category', 'sub_child_category', 'images_paths', 'description',
51
+ 'quantity', 'promotion_on_the_pack', 'type_of_packaging', 'mrp'])
52
+
53
 
54
 
55
  submit_button_tab_1.click(fn=sample_fun,inputs=[voice_input,prodcut_id])
56
+ submit_button_tab_4.click(fn=search_products,inputs=[embbed_text_search] ,outputs= dataframe_output_tab_4)
57
 
58
  demo.launch()