Added gemini pro

.env

@@ -0,0 +1 @@
+GOOGLE_API_KEY="AIzaSyD5k4WkMPdP-SkmRgswFm-6OjqSv8U1_Zw"

app.py

@@ -10,15 +10,34 @@ import requests
 from io import BytesIO
 from bs4 import BeautifulSoup
 from urllib.parse import urljoin
+import google.generativeai as genai
+from langchain_google_genai import GoogleGenerativeAIEmbeddings
+from langchain_google_genai import ChatGoogleGenerativeAI
+import warnings
+from dotenv import load_dotenv
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 app = Flask(__name__)
 cors = CORS(app)
+load_dotenv()
+
+# # Define the model and feature extractor globally
+# model = AutoModelForImageClassification.from_pretrained('carbon225/vit-base-patch16-224-hentai')
+# feature_extractor = AutoFeatureExtractor.from_pretrained('carbon225/vit-base-patch16-224-hentai')
+
+def load_model():  
+  api_key=os.getenv("GOOGLE_API_KEY")
+  genai.configure(api_key=api_key)
+  model = ChatGoogleGenerativeAI(model="gemini-pro",
+                             temperature=0.3)
+  
+  return model
+
+def load_embeddings():
+  embeddings = GoogleGenerativeAIEmbeddings(model = "models/embedding-001")
 
-# Define the model and feature extractor globally
-model = AutoModelForImageClassification.from_pretrained('carbon225/vit-base-patch16-224-hentai')
-feature_extractor = AutoFeatureExtractor.from_pretrained('carbon225/vit-base-patch16-224-hentai')
+  return embeddings
 
 @app.route("/", methods=["GET"])
 def default():
@@ -40,38 +59,50 @@ def extract_images():
     except Exception as e:
         return e
 
-@app.route("/predict", methods=["GET"])
-def predict():
-    try:
-        src = request.args.get("src")
+# @app.route("/predict", methods=["GET"])
+# def predict():
+#     try:
+#         src = request.args.get("src")
 
-        # Download image from the provided URL
-        response = requests.get(src)
-        response.raise_for_status()  
+#         # Download image from the provided URL
+#         response = requests.get(src)
+#         response.raise_for_status()  
 
-        # Open and preprocess the image
-        image = Image.open(BytesIO(response.content))
-        image = image.resize((128, 128))
+#         # Open and preprocess the image
+#         image = Image.open(BytesIO(response.content))
+#         image = image.resize((128, 128))
 
-        # Extract features using the pre-trained feature extractor
-        encoding = feature_extractor(images=image.convert("RGB"), return_tensors="pt")
+#         # Extract features using the pre-trained feature extractor
+#         encoding = feature_extractor(images=image.convert("RGB"), return_tensors="pt")
 
-        # Make a prediction using the pre-trained model
-        with torch.no_grad():
-            outputs = model(**encoding)
-            logits = outputs.logits
+#         # Make a prediction using the pre-trained model
+#         with torch.no_grad():
+#             outputs = model(**encoding)
+#             logits = outputs.logits
 
-        # Get the predicted class index and label
-        predicted_class_idx = logits.argmax(-1).item()
-        predicted_class_label = model.config.id2label[predicted_class_idx]
+#         # Get the predicted class index and label
+#         predicted_class_idx = logits.argmax(-1).item()
+#         predicted_class_label = model.config.id2label[predicted_class_idx]
 
-        # Return the predictions
-        return json.dumps({"class": predicted_class_label})
+#         # Return the predictions
+#         return json.dumps({"class": predicted_class_label})
 
-    except requests.exceptions.RequestException as e:
-        return json.dumps({"error": f"Request error: {str(e)}"})
-    except Exception as e:
-        return json.dumps({"error": f"An unexpected error occurred: {str(e)}"})
+#     except requests.exceptions.RequestException as e:
+#         return json.dumps({"error": f"Request error: {str(e)}"})
+#     except Exception as e:
+#         return json.dumps({"error": f"An unexpected error occurred: {str(e)}"})
+    
+@app.route('/answer',methods=['POST'])
+def answer():
+  api_key=os.getenv("GOOGLE_API_KEY")
+  genai.configure(api_key=api_key)
+  model=genai.GenerativeModel('gemini-pro')
+  query=request.get_json()['query']
+  final_query=f"""
+  Following are negative reviews about my products, suggest what are the key issues from the customer feedback:{query}
+  """
+  response = model.generate_content(final_query)
+  return json.dumps({"message":response.text})
 
 if __name__ == "__main__":
     app.run(debug=True)

requirements.txt

@@ -1,22 +1,74 @@
+affine==2.4.0
+aiohttp==3.9.3
+aiosignal==1.3.1
+alabaster==0.7.16
+albumentations==1.4.1
+annotated-types==0.6.0
+anyio==4.3.0
+attrs==23.2.0
+Babel==2.14.0
+beautifulsoup4==4.12.3
 blinker==1.7.0
-bs4==0.0.1
-certifi==2023.11.17
+cachetools==5.3.3
+certifi==2024.2.2
 charset-normalizer==3.3.2
 click==8.1.7
+click-plugins==1.1.1
+cligj==0.7.2
+commonmark==0.9.1
+contourpy==1.2.0
+cycler==0.12.1
+dataclasses-json==0.6.4
+deepforest @ file:///home/hawkeye/Desktop/DeepForest
+distlib==0.3.8
+docutils==0.20.1
 filelock==3.13.1
-Flask==3.0.0
+fiona==1.9.5
+Flask==3.0.2
 Flask-Cors==4.0.0
-fsspec==2023.12.2
-huggingface-hub==0.20.1
+fonttools==4.49.0
+frozenlist==1.4.1
+fsspec==2024.2.0
+geopandas==0.14.3
+google==3.0.0
+google-ai-generativelanguage==0.4.0
+google-api-core==2.18.0
+google-auth==2.25.1
+google-generativeai==0.4.1
+googleapis-common-protos==1.63.0
+greenlet==3.0.3
+grpcio==1.62.1
+grpcio-status==1.62.1
 idna==3.6
-importlib-metadata==7.0.0
+imagecodecs==2024.1.1
+imageio==2.34.0
+imagesize==1.4.1
+insta-scrape==2.1.2
+instagramy==4.5
+instaloader==4.11
 itsdangerous==2.1.2
-Jinja2==3.1.2
-markdownify==0.11.6
-MarkupSafe==2.1.3
+Jinja2==3.1.3
+joblib==1.3.2
+jsonpatch==1.33
+jsonpointer==2.4
+kiwisolver==1.4.5
+langchain==0.1.13
+langchain-community==0.0.29
+langchain-core==0.1.33
+langchain-google-genai==0.0.11
+langchain-text-splitters==0.0.1
+langsmith==0.1.31
+lazy_loader==0.3
+lightning-utilities==0.10.1
+MarkupSafe==2.1.5
+marshmallow==3.21.1
+matplotlib==3.8.3
+mock==5.1.0
 mpmath==1.3.0
+multidict==6.0.5
+mypy-extensions==1.0.0
 networkx==3.2.1
-numpy==1.26.2
+numpy==1.26.4
 nvidia-cublas-cu12==12.1.3.1
 nvidia-cuda-cupti-cu12==12.1.105
 nvidia-cuda-nvrtc-cu12==12.1.105
@@ -26,33 +78,74 @@ nvidia-cufft-cu12==11.0.2.54
 nvidia-curand-cu12==10.3.2.106
 nvidia-cusolver-cu12==11.4.5.107
 nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.18.1
-nvidia-nvjitlink-cu12==12.3.101
+nvidia-nccl-cu12==2.19.3
+nvidia-nvjitlink-cu12==12.4.99
 nvidia-nvtx-cu12==12.1.105
-outcome==1.2.0
+opencv-python==4.9.0.80
+opencv-python-headless==4.9.0.80
+orjson==3.9.15
 packaging==23.2
-perplexity-api @ git+https://gitlab.com/robowaifudev/perplexity-api@0baebed39beeedc13e6281af5505f4c4221fdb5f
-Pillow==10.1.0
-PySocks==1.7.1
+pandas==2.2.1
+pillow==10.2.0
+platformdirs==4.2.0
+progressbar2==4.4.2
+proto-plus==1.23.0
+protobuf==4.25.3
+psutil==5.9.8
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pycocotools==2.0.7
+pydantic==2.6.4
+pydantic_core==2.16.3
+Pygments==2.17.2
+pyparsing==3.1.2
+pyproj==3.6.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-utils==3.8.2
+pytorch-lightning==2.2.1
+pytz==2024.1
 PyYAML==6.0.1
-regex==2023.10.3
+rasterio==1.3.9
+readthedocs-build==2.0.9
+readthedocs-sphinx-ext==2.2.5
+recommonmark==0.7.1
 requests==2.31.0
-safetensors==0.4.1
-selenium==4.12.0
-sortedcontainers==2.4.0
+rsa==4.9
+Rtree==1.2.0
+scikit-image==0.22.0
+scikit-learn==1.4.1.post1
+scipy==1.12.0
+setuptools==68.2.2
+shapely==2.0.3
+six==1.16.0
+slidingwindow==0.0.14
+sniffio==1.3.1
+snowballstemmer==2.2.0
+snuggs==1.4.7
+soupsieve==2.5
+Sphinx==7.2.6
+sphinxcontrib-applehelp==1.0.8
+sphinxcontrib-devhelp==1.0.6
+sphinxcontrib-htmlhelp==2.0.5
+sphinxcontrib-jsmath==1.0.1
+sphinxcontrib-qthelp==1.0.7
+sphinxcontrib-serializinghtml==1.1.10
+SQLAlchemy==2.0.28
 sympy==1.12
-tokenizers==0.15.0
-torch==2.1.2
-tqdm==4.66.1
-transformers==4.36.2
-trio==0.22.2
-trio-websocket==0.10.4
-triton==2.1.0
-typing_extensions==4.9.0
-urllib3==2.1.0
+tenacity==8.2.3
+threadpoolctl==3.3.0
+tifffile==2024.2.12
+torch==2.2.1
+torchmetrics==1.3.1
+torchvision==0.17.1
+tqdm==4.66.2
+typing-inspect==0.9.0
+typing_extensions==4.10.0
+tzdata==2024.1
+urllib3==2.2.1
+virtualenv==20.25.1
 Werkzeug==3.0.1
-wsproto==1.2.0
-zipp==3.17.0
-gunicorn
-torch
-bs4
+wheel==0.41.2
+xmltodict==0.13.0
+yarl==1.9.4