Upload embedding.py

embedding.py

@@ -78,7 +78,6 @@ def imporve_text(text):
     Please rewrite the following text to make it short, concise, and of high quality.
     Ensure that all essential information and key points are retained. 
     Focus on improving clarity, coherence, and word choice without altering the original meaning.
-    Do not add your own information or titles.
 
     text = {text}
     '''
@@ -111,23 +110,15 @@ def feature_extraction(tag, history , context):
     return result.content
 
 def feature_extraction_image(url):
-
-    vision = ChatGoogleGenerativeAI(model="gemini-1.5-flash",google_api_key='AIzaSyBzbZQBffHFK3N-gWnhDDNbQ9yZnZtaS2E',temperature = 0.1)
-    # result = gemini.invoke('''Hello''')
-    # Markdown(result.content)
-    # print(result)
-
-    text = 'None'
-    message = HumanMessage(content=[
-                    {"type": "text", "text": "Please, Describe this image in detail"},
-                    {"type": "image_url", "image_url": url}
-                ])
+    text = ' '
+    model = genai.GenerativeModel('gemini-1.5-flash-001')
     try:
-        model = random.choice([vision,vision1,vision2,vision3])
-        text = model.invoke([message])
+        res = model.generate_content(['Describe this image to me',url])
+        text = res.text
+
     except:
-        return text
-    return text.content
+        pass
+    return text
 
 def detailed_feature_extraction(find, context):
 
@@ -375,4 +366,5 @@ text_splitter_small = RecursiveCharacterTextSplitter(
 )
 
 if __name__ == '__main__':
-    pass
+    print(get_embed_chroma('https://www.galaxys24manual.com/wp-content/uploads/pdf/galaxy-s24-manual-SAM-S921-S926-S928-OS14-011824-FINAL-US-English.pdf'))
+    # print(get_image_embeddings(Product='Samsung Galaxy S24'))