changes

Dockerfile

@@ -10,6 +10,6 @@ RUN python3.11 -m pip install --no-cache-dir -r requirements.txt
 COPY . .
 ENTRYPOINT ["/bin/sh"]
 ENV FLASK_APP=ddl_query_generator.py
-CMD ["/usr/bin/python3.11", "flask", "run", "ddl_query_generator.py", "--host=0.0.0.0", "--port=7860"]
+CMD ["/usr/bin/python3.11", "flask", "run", "--host=0.0.0.0", "--port=7860"]
 # CMD ["/usr/bin/python3.11", "ddl_query_generator.py"]
 # flask run ddl_query_generator.py

ddl_query_generator.py

@@ -4,20 +4,26 @@ import warnings
 import pandas as pd
 from flask import Flask, render_template, request
 from langchain_community.llms import Ollama
-from langchain import PromptTemplate, LLMChain
+from langchain_core.prompts import PromptTemplate
+from langchain.chains import LLMChain
 
+print("Starting the server...")
 app = Flask(__name__)
 
 # Suppressing warnings
 warnings.filterwarnings("ignore")
 
 # Initializing the language model
+
+# Ollama model
+print("Initializing the language model...")
 llm = Ollama(model="pxlksr/defog_sqlcoder-7b-2:Q4_K")
 
 UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
 # Prompt template for language model
+print("Initializing the prompt template...")
 template = """
 ### Task
 Generate a SQL query to answer [QUESTION]{user_question}[/QUESTION]
@@ -58,6 +64,7 @@ def run_sql_query(csv_files, sql_query):
 history = []
 history_dll = []
 
+print("Server started successfully!")
 @app.route('/', methods=['GET', 'POST'])
 def ddl_query():
     ddl = None
@@ -97,4 +104,5 @@ def index():
     return render_template('database_selection_index.html', result=result, query=query, history=history)
 
 if __name__ == '__main__':
+    print("Running the server...")
     app.run(debug=True, port = 7860, host = '0.0.0.0')

wsgi.py

@@ -0,0 +1,4 @@
+from ddl_query_generator import app
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', port=7860, debug=True)