Upload 20 files

Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.11
+USER root
+
+WORKDIR /app
+RUN rm -f /etc/apt/apt.conf.d/docker-clean \
+    && echo 'Binary::apt::APT::Keep-Downloaded-Packages "true";' > /etc/apt/apt.conf.d/keep-cache
+
+
+RUN curl -sSL https://install.python-poetry.org | python3 -
+
+ENV POETRY_HOME="/root/.local"
+ENV PATH="$POETRY_HOME/bin:$PATH"
+COPY pyproject.toml poetry.lock* /app/
+
+RUN poetry config virtualenvs.create false && poetry install --no-interaction --no-ansi --no-root
+# Copy project files
+
+ADD ./app /app/app
+ADD ./utils /app/utils
+COPY main.py /app/main.py
+
+CMD ["uvicorn", "main:app", "--host 0.0.0.0", "--port", "8000", "--reload"]

alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration.

alembic/env.py

@@ -0,0 +1,55 @@
+import os
+from dotenv import load_dotenv
+from alembic import context
+from sqlalchemy import engine_from_config, pool
+from app.models import Base
+
+
+load_dotenv()
+
+# Alembic Config object
+config = context.config
+
+db_url = os.getenv("POSTGRES_DATABASE_URL")
+if not db_url:
+    raise RuntimeError("POSTGRES_DATABASE_URL not set in the environment!")
+
+
+config.set_main_option("sqlalchemy.url", db_url)
+target_metadata = Base.metadata
+
+def run_migrations_online():
+    """Run migrations in 'online' mode."""
+    # Create an engine from the configuration
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        # Configure the context with the connection and the metadata
+        context.configure(
+            connection=connection, target_metadata=target_metadata
+        )
+
+        with context.begin_transaction():
+            context.run_migrations()
+
+def run_migrations_offline():
+    """Run migrations in 'offline' mode."""
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

alembic/script.py.mako

@@ -0,0 +1,26 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

alembic/versions/8dc44341a607_initial_migration.py

@@ -0,0 +1,41 @@
+"""Initial migration
+
+Revision ID: 8dc44341a607
+Revises: 
+Create Date: 2025-02-20 23:19:04.480316
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '8dc44341a607'
+down_revision: Union[str, None] = None
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('users',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(), nullable=True),
+    sa.Column('sender_number', sa.String(), nullable=True),
+    sa.Column('chat_history', sa.JSON(), nullable=True),
+    sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_users_id'), 'users', ['id'], unique=False)
+    op.create_index(op.f('ix_users_sender_number'), 'users', ['sender_number'], unique=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f('ix_users_sender_number'), table_name='users')
+    op.drop_index(op.f('ix_users_id'), table_name='users')
+    op.drop_table('users')
+    # ### end Alembic commands ###

app/crud.py

@@ -0,0 +1,21 @@
+from app import models
+from sqlalchemy.orm import Session
+
+
+def get_user_by_sender_number(db: Session, sender_number: str):
+    return db.query(models.User).filter(models.User.sender_number == sender_number).first()
+
+def get_user_by_user_id(db: Session, user_id: int):
+    return db.query(models.User).filter(models.User.id == user_id).first()
+
+def create_user(db: Session, name: str, sender_number: str):
+    try:
+        new_user = models.User(name=name, sender_number=sender_number)
+        db.add(new_user)
+        db.commit()
+        db.refresh(new_user)
+        return new_user
+    except Exception as e:
+        db.rollback()
+        print(f"Error creating user: {e}")
+        return None

app/database.py

@@ -0,0 +1,14 @@
+import os
+from dotenv import load_dotenv
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.declarative import declarative_base
+
+load_dotenv()
+
+SQLALCHEMY_DATABASE_URL = os.getenv("POSTGRES_DATABASE_URL")
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()

app/models.py

@@ -0,0 +1,11 @@
+from sqlalchemy import Column, Integer, JSON, String, DateTime, func
+from .database import Base
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, default="Unknown")
+    sender_number = Column(String, unique=True, index=True)
+    chat_history = Column(JSON, default=[])
+    created_at = Column(DateTime(timezone=True), server_default=func.now())

app/router.py

@@ -0,0 +1,144 @@
+
+from app import models, crud
+from app.database import SessionLocal, engine
+from utils.log_utils import setup_logger
+from fastapi import Request, Depends, APIRouter, HTTPException
+import os, json, random, requests
+from livekit import api
+from sqlalchemy.orm import Session
+from fastapi.responses import JSONResponse
+from twilio.twiml.messaging_response import MessagingResponse
+from fastapi.responses import Response
+from pydantic import BaseModel, Field
+from dotenv import load_dotenv
+load_dotenv()
+
+logger = setup_logger("router")
+router = APIRouter()
+
+models.Base.metadata.create_all(bind=engine)
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+
+class MessageRequestValidate(BaseModel):
+    from_phone: str = Field(default="+917389058485")
+
+@router.post('/realtor/validate')
+async def realtor_validate_user(request: MessageRequestValidate, db: Session = Depends(get_db)):
+    logger.info(f"request: {request}")
+    user = crud.get_user_by_sender_number(db, sender_number=request.from_phone)
+    if user:
+        return JSONResponse(content="User is verified")
+    return JSONResponse(content="user is not verified")
+
+class MessageRequestSignUP(BaseModel):
+    from_phone: str = Field(default="+917389058485")
+    agent_id: str = Field(default="agent_id1")
+    full_name: str = Field(default="agent_full_name")
+    broker_id: str = Field(default="broker_id1")
+
+@router.post('/realtor/singup')
+async def realtor_sign_up(request: MessageRequestSignUP):
+    print(request)
+    return JSONResponse(content="user is signed up sucessfully now you can contine to ask propertry related questions")
+
+
+@router.post("/create_user")
+async def create_user(name: str, number: str, db: Session = Depends(get_db)):
+    user = crud.create_user(db, name=name, sender_number=number)
+    if not user:
+        raise HTTPException(status_code=400, detail="User creation failed")
+    return user
+
+@router.post('/chatbot/')
+async def whatsapp_webhook(request: Request):
+    form_data = await request.form()
+
+    num_media = int(form_data.get('NumMedia', 0))
+    sender_number = form_data.get("From")[9:]
+    profile_name = form_data.get("ProfileName")
+    user_query = form_data.get("Body")
+
+    logger.warning(f"num_media: {num_media}, sender_number: {sender_number}, ProfileName: {profile_name}, user_query: {user_query}")
+
+    payload = {
+        "input_value": json.dumps({"from_phone": sender_number, "msg": user_query}),
+        "session_id": sender_number
+    }
+    logger.info(f"Payload: {payload}")
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    url = "https://workflows.kickcall.ai/api/v1/run/c94b14f5-d7a7-4fc8-89d8-eb672716c778?stream=false&user_id=1"
+
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(payload))
+        logger.info(f"[Langflow][run] Response status code received: {response.status_code}")
+
+        if response.status_code == 200:
+            response_json = response.json()
+            logger.info(f"[Langflow][run] Response: {response_json}")
+            output_text = response_json["outputs"][0]["outputs"][0]["results"]["message"]["data"]["text"]
+
+            bot_resp = MessagingResponse()
+            msg = bot_resp.message()
+            msg.body(output_text)
+
+            return Response(content=str(bot_resp), media_type="application/xml")
+        else:
+            logger.info(f"Error: Received status code {response.status_code}")
+            bot_resp = MessagingResponse()
+            msg = bot_resp.message()
+            msg.body("I'm sorry, I couldn't process your request at the moment. Please try again later.")
+            return Response(content=str(bot_resp), media_type="application/xml")
+
+    except Exception as e:
+        logger.info(f"Error in processing user query: {e}")
+        bot_resp = MessagingResponse()
+        msg = bot_resp.message()
+        msg.body("I'm sorry, an unexpected error occurred. Please try again later.")
+        return Response(content=str(bot_resp), media_type="application/xml")
+
+
+class MessageRequestRunFlow(BaseModel):
+    from_phone: str = Field(default="+917389058485")
+    user_query: str = Field(default="Hello")
+
+@router.post('/realtor/run_flow')
+async def realtor_validate_user(request: MessageRequestRunFlow):
+    print(f"body: {request}")
+    user_query = request.user_query
+    print(f"user_query: {user_query}")
+    payload = {
+        "input_value": json.dumps({"from_phone": request.from_phone, "msg": user_query}),
+        "session_id": request.from_phone
+    }
+    print(f"Payload: {payload}")
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    url = "https://workflows.kickcall.ai/api/v1/run/c94b14f5-d7a7-4fc8-89d8-eb672716c778?stream=false&user_id=1"
+
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(payload))
+        print(f"[Langflow][run] Response status code received: {response.status_code}")
+
+        if response.status_code == 200:
+            response_json = response.json()
+            print(f"[Langflow][run] Response: {response_json}")
+            output_text = response_json["outputs"][0]["outputs"][0]["results"]["message"]["data"]["text"]
+            return JSONResponse(content=output_text)
+        else:
+            print(f"Error: Received status code {response.status_code}")
+            msg ="I'm sorry, I couldn't process your request at the moment. Please try again later."
+            return JSONResponse(content=msg)
+    except Exception as e:
+        print(f"Error in processing user query: {e}")
+        msg = "I'm sorry, an unexpected error occurred. Please try again later."
+        return JSONResponse(content=msg)

main.py

@@ -0,0 +1,7 @@
+from fastapi import FastAPI
+from app.router import router
+
+
+app = FastAPI()
+
+app.include_router(router)

pyproject.toml

@@ -0,0 +1,24 @@
+[tool.poetry]
+name = "bluebash-test1"
+version = "0.1.0"
+description = ""
+authors = ["narendra-bluebash <[email protected]>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.12"
+fastapi = "^0.115.6"
+sqlalchemy = "^2.0.37"
+python-dotenv = "^1.0.1"
+uvicorn = "^0.34.0"
+colorlog = "^6.9.0"
+python-multipart = "^0.0.20"
+twilio = "^9.4.2"
+alembic = "^1.14.0"
+livekit = "^0.20.1"
+livekit-api = "^0.8.2"
+psycopg2 = "^2.9.10"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

utils/log_utils.py

@@ -0,0 +1,26 @@
+import logging
+import colorlog
+
+def setup_logger(logger_name: str, debug_color="white", info_color="green", propagate=False):
+    color_handler = colorlog.StreamHandler()
+    log_colors = {
+        "DEBUG": debug_color,
+        "INFO": info_color,
+        "WARNING": "yellow",
+        "ERROR": "red",
+        "CRITICAL": "bold_red",
+    }
+    color_handler.setFormatter(
+        colorlog.ColoredFormatter(
+            "%(log_color)s%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+            datefmt="%Y-%m-%d %H:%M:%S",
+            log_colors=log_colors
+        )
+    )
+    logger = logging.getLogger(logger_name)
+    if logger.hasHandlers():
+        logger.handlers.clear()
+    logger.setLevel(logging.DEBUG)
+    logger.addHandler(color_handler)
+    logger.propagate = propagate
+    return logger