Notes

os.environ['GOOGLE_API_KEY'] = "AIzaSyDtj5XMeoze2InmywR824OWKvTYGzp71o4"
class GenerateContentView(APIView):

'''
class for generte the content based on a given prompt
'''

def post(self, request):
'''
It is a post API for this Class
'''
content = request.data.get('content', '')
category = request.data.get('category','')
print(content,category,'------')

if not content:
return Response({"Message": "Content is required"}, status=status.HTTP_400_BAD_REQUEST)

elif not category:
return Response({"Message": "Category is required"}, status=status.HTTP_400_BAD_REQUEST)

if not is_valid_url(content):
return Response({"Message": "Invalid URL"}, status=status.HTTP_400_BAD_REQUEST)

else:

api_key = os.environ['GOOGLE_API_KEY']
genai.configure(api_key = api_key)
model = genai.GenerativeModel('gemini-pro')

if category == 'Blogs':
response = model.generate_content(
f"Generate Informative, In-Depth, Structured, Feature Call to Action at conclusion and also summarize blog in 800-1200 words from this URL {content}"
)
elif category == 'Webpages':
response = model.generate_content(
f"Generate Concise, informative, active language, structured (H2s and H3s), sales-oriented with some technical expertise, 1500-2000 words from this URL {content}"
)
elif category == 'Case Studies and White Papers':
response = model.generate_content(
f"Generate Informative, Technical, Solution-Oriented, 500-1000 words URL {content}"
)
elif category == 'Flyers':
response = model.generate_content(
f" generate tge concise, well-structured, sales-oriented but conveying technical expertise for this url {content}"
)


else:
return Response({"Message": "Please enter a valid category"}, status=status.HTTP_400_BAD_REQUEST)

try:

json_response = response.text
text_document = ContentResponse()
text_document.document.save('document.txt', ContentFile(json_response))

print('********* n ',content,'n************')
print(json_response, 'n -----------------------')

return Response({
"status": status.HTTP_200_OK,
"message": "Success",
"data": {
"Content": json_response
}
}, status=status.HTTP_201_CREATED)

except Exception as e:

return Response({
"status": status.HTTP_500_INTERNAL_SERVER_ERROR,
"message": "Error occurred while generating Content Please check the Provided data."
}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)

def is_valid_url(url):
try:
result = urlparse(url)
return all([result.scheme, result.netloc]) and '.' in result.netloc
except ValueError:
return False