pvanand commited on
Commit
b8812f4
·
verified ·
1 Parent(s): 466d5bb

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +13 -2
main.py CHANGED
@@ -10,6 +10,8 @@ from fastapi import Query
10
  from pydantic import BaseModel
11
  from fastapi.middleware.cors import CORSMiddleware
12
  from helper_functions_api import md_to_html
 
 
13
 
14
  # Retrieve environment variables
15
  TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
@@ -66,7 +68,12 @@ def generate_report(topic, description):
66
  md_report = together_response(prompt, model = "meta-llama/Llama-3-70b-chat-hf", SysPrompt = SysPromptMdOffline)
67
  return md_to_html(md_report)
68
 
69
-
 
 
 
 
 
70
  # Define the app
71
  app = FastAPI()
72
 
@@ -101,4 +108,8 @@ async def create_topics(input: TopicInput):
101
  async def create_report(input: ReportInput):
102
  report = generate_report(input.topic, input.description) # You'll need to implement this function
103
  return {"report": report}
104
-
 
 
 
 
 
10
  from pydantic import BaseModel
11
  from fastapi.middleware.cors import CORSMiddleware
12
  from helper_functions_api import md_to_html
13
+ from duckduckgo_search import DDGS
14
+ import time
15
 
16
  # Retrieve environment variables
17
  TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
 
68
  md_report = together_response(prompt, model = "meta-llama/Llama-3-70b-chat-hf", SysPrompt = SysPromptMdOffline)
69
  return md_to_html(md_report)
70
 
71
+ def get_images(query, num_results):
72
+ time.sleep(0.5)
73
+ ddgs = DDGS()
74
+ imgs = ddgs.images(keywords=query, safesearch="on", max_results=num_results)
75
+ return imgs
76
+
77
  # Define the app
78
  app = FastAPI()
79
 
 
108
  async def create_report(input: ReportInput):
109
  report = generate_report(input.topic, input.description) # You'll need to implement this function
110
  return {"report": report}
111
+
112
+ @app.post("/get_images")
113
+ async def get_images(input: ReportInput):
114
+ images = get_images(input.user_input, input.num_topics)
115
+ return {"images": images}